Home
last modified time | relevance | path

Searched refs:src_dt (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/debug/
H A Ddebug_graph_utils.cc170 const DataType src_dt = src_node->output_type(src_output_slot); in InsertNodes() local
185 src_node->name(), src_output_slot, src_dt, tensor_name, in InsertNodes()
204 src_dt, tensor_name, debug_urls, i, in InsertNodes()
301 const string& src_node_name, const int src_output, const DataType src_dt, in CreateCopyNode() argument
334 .Input(src_node_name, src_output, src_dt) in CreateCopyNode()
473 const DataType src_dt, const string& tensor_name, in CreateDebugNode() argument
487 .Input(src_copy_node_name, 0, src_dt) in CreateDebugNode()
H A Ddebug_graph_utils.h96 const string& src_node_name, const int src_output, const DataType src_dt,
115 const DataType src_dt,
/aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/
H A DCpuCastKernel.cpp54 …[](const CastDataTypeISASelectorData & data) { return data.src_dt == DataType::QASYMM8_SIGNED && d… in __anon2cc4acfe0202()
59 …[](const CastDataTypeISASelectorData & data) { return data.src_dt == DataType::QASYMM8 && data.dst… in __anon2cc4acfe0302()
64 …[](const CastDataTypeISASelectorData & data) { return data.src_dt == DataType::U8 && data.dst_dt =… in __anon2cc4acfe0402()
69 …[](const CastDataTypeISASelectorData & data) { return data.src_dt == DataType::F16 && data.isa.fp1… in __anon2cc4acfe0502()
74 …[](const CastDataTypeISASelectorData & data) { return data.src_dt == DataType::F32 && data.dst_dt … in __anon2cc4acfe0602()
79 …[](const CastDataTypeISASelectorData & data) { return data.src_dt == DataType::F32 && data.dst_dt … in __anon2cc4acfe0702()
84 …[](const CastDataTypeISASelectorData & data) { return data.src_dt == DataType::S32 && data.dst_dt … in __anon2cc4acfe0802()
89 …[](const CastDataTypeISASelectorData & data) { return data.src_dt == DataType::BFLOAT16 && data.ds… in __anon2cc4acfe0902()
H A DCpuKernelSelectionTypes.h52 DataType src_dt; member
/aosp_15_r20/external/pytorch/test/distributed/_tensor/
H A Dtest_tensor_ops.py414 src_dt = distribute_tensor(global_src, device_mesh, [Replicate()])
416 src_dt = global_src
421 output_dt = torch.scatter(input_dt, scatter_dim, index_dt, src_dt)
/aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/
H A DConvolutionLayerFixture.h317 const DataType src_dt = _is_bfloat16 ? DataType::F32 : _data_type; variable
322 SimpleTensor<T> src{ input_shape, src_dt, 1, _quantization_info };