Searched refs:kComplex64 (Results 1 – 18 of 18) sorted by relevance
/external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
D | resolve_multiply_by_zero.cc | 142 case ArrayDataType::kComplex64: { in Run() 144 constant_input_array.GetBuffer<ArrayDataType::kComplex64>().data; in Run() 145 if (!AreAllBufferElementsZero<DataType<ArrayDataType::kComplex64>>( in Run() 149 FillArrayWithZeros<ArrayDataType::kComplex64>(&output_array); in Run()
|
D | resolve_constant_pack.cc | 104 case ArrayDataType::kComplex64: in Run() 105 Pack<ArrayDataType::kComplex64>(model, *op); in Run()
|
D | resolve_constant_reshape.cc | 98 case ArrayDataType::kComplex64: in Run() 99 CopyArrayBuffer<ArrayDataType::kComplex64>(input_array, &output_array); in Run()
|
D | resolve_constant_fill.cc | 101 case ArrayDataType::kComplex64: in Run() 102 if (!ComputeFillArray<ArrayDataType::kComplex64>(model, op)) { in Run()
|
D | resolve_constant_strided_slice.cc | 156 case ArrayDataType::kComplex64: in Run() 157 StridedSlice<ArrayDataType::kComplex64>(*op, input_array, &output_array); in Run()
|
D | resolve_constant_gather.cc | 135 case ArrayDataType::kComplex64: in Run() 136 Gather<ArrayDataType::kComplex64>(input_array, coords_array, in Run()
|
D | resolve_constant_tile.cc | 153 case ArrayDataType::kComplex64: in Run() 154 Tile<ArrayDataType::kComplex64>(input_array, multiples_array, in Run()
|
D | resolve_constant_slice.cc | 150 case ArrayDataType::kComplex64: in Run() 151 if (!Slice<ArrayDataType::kComplex64>(*op, input_array, &output_array)) { in Run()
|
D | resolve_constant_transpose.cc | 162 case ArrayDataType::kComplex64: in Run() 163 Transpose<ArrayDataType::kComplex64>(model, input_array, op->perm, in Run()
|
D | resolve_constant_concatenation.cc | 201 case ArrayDataType::kComplex64: in Run() 202 ConcatenateTensorBuffers<ArrayDataType::kComplex64>( in Run()
|
D | quantization_util.cc | 121 case ArrayDataType::kComplex64: in ChooseQuantizationParamsForArrayAndQuantizedDataType()
|
/external/tensorflow/tensorflow/lite/toco/tflite/ |
D | types.cc | 103 case ArrayDataType::kComplex64: in Serialize() 129 return ArrayDataType::kComplex64; in Deserialize() 154 case ArrayDataType::kComplex64: in Serialize() 155 return CopyBuffer<ArrayDataType::kComplex64>(array, builder); in Serialize() 182 return CopyBuffer<ArrayDataType::kComplex64>(buffer, array); in Deserialize()
|
D | types_test.cc | 77 {ArrayDataType::kComplex64, ::tflite::TensorType_COMPLEX64}}; in TEST() 178 Array recovered = ToFlatBufferAndBack<ArrayDataType::kComplex64>( in TEST() 180 EXPECT_THAT(recovered.GetBuffer<ArrayDataType::kComplex64>().data, in TEST()
|
/external/tensorflow/tensorflow/lite/toco/ |
D | tooling_util.cc | 90 case ArrayDataType::kComplex64: in ArrayDataTypeName() 822 case ArrayDataType::kComplex64: in CompareConstantArrays() 823 return CompareArrayBuffers<ArrayDataType::kComplex64>(lhs_array, in CompareConstantArrays() 1435 case ArrayDataType::kComplex64: in CloneArray() 1436 CopyArrayBuffer<ArrayDataType::kComplex64>(source_array, &target_array); in CloneArray() 1751 case ArrayDataType::kComplex64: in ElementSize() 2273 return ArrayDataType::kComplex64; in ConvertIODataTypeToArrayDataType()
|
D | model.h | 222 kComplex64, enumerator 277 struct DataTypeImpl<ArrayDataType::kComplex64> {
|
D | import_tensorflow_test.cc | 355 constexpr ArrayDataType kType = ArrayDataType::kComplex64; in TEST_F()
|
D | export_tensorflow.cc | 66 case ArrayDataType::kComplex64: in GetTensorFlowDataType() 302 const auto& data = array.GetBuffer<ArrayDataType::kComplex64>().data; in ConvertComplex64TensorConst() 2409 case ArrayDataType::kComplex64: in AddPlaceholder() 2471 case ArrayDataType::kComplex64: in ExportTensorFlowGraphDefImplementation()
|
D | import_tensorflow.cc | 191 return ArrayDataType::kComplex64; in ConvertDataType() 392 output_array->GetMutableBuffer<ArrayDataType::kComplex64>().data; in ImportComplex64Array() 601 array.data_type = ArrayDataType::kComplex64; in ConvertConstOperator()
|