Home
last modified time | relevance | path

Searched refs:kComplex64 (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dresolve_multiply_by_zero.cc142 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()
Dresolve_constant_pack.cc104 case ArrayDataType::kComplex64: in Run()
105 Pack<ArrayDataType::kComplex64>(model, *op); in Run()
Dresolve_constant_reshape.cc98 case ArrayDataType::kComplex64: in Run()
99 CopyArrayBuffer<ArrayDataType::kComplex64>(input_array, &output_array); in Run()
Dresolve_constant_fill.cc101 case ArrayDataType::kComplex64: in Run()
102 if (!ComputeFillArray<ArrayDataType::kComplex64>(model, op)) { in Run()
Dresolve_constant_strided_slice.cc156 case ArrayDataType::kComplex64: in Run()
157 StridedSlice<ArrayDataType::kComplex64>(*op, input_array, &output_array); in Run()
Dresolve_constant_gather.cc135 case ArrayDataType::kComplex64: in Run()
136 Gather<ArrayDataType::kComplex64>(input_array, coords_array, in Run()
Dresolve_constant_tile.cc153 case ArrayDataType::kComplex64: in Run()
154 Tile<ArrayDataType::kComplex64>(input_array, multiples_array, in Run()
Dresolve_constant_slice.cc150 case ArrayDataType::kComplex64: in Run()
151 if (!Slice<ArrayDataType::kComplex64>(*op, input_array, &output_array)) { in Run()
Dresolve_constant_transpose.cc162 case ArrayDataType::kComplex64: in Run()
163 Transpose<ArrayDataType::kComplex64>(model, input_array, op->perm, in Run()
Dresolve_constant_concatenation.cc201 case ArrayDataType::kComplex64: in Run()
202 ConcatenateTensorBuffers<ArrayDataType::kComplex64>( in Run()
Dquantization_util.cc121 case ArrayDataType::kComplex64: in ChooseQuantizationParamsForArrayAndQuantizedDataType()
/external/tensorflow/tensorflow/lite/toco/tflite/
Dtypes.cc103 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()
Dtypes_test.cc77 {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/
Dtooling_util.cc90 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()
Dmodel.h222 kComplex64, enumerator
277 struct DataTypeImpl<ArrayDataType::kComplex64> {
Dimport_tensorflow_test.cc355 constexpr ArrayDataType kType = ArrayDataType::kComplex64; in TEST_F()
Dexport_tensorflow.cc66 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()
Dimport_tensorflow.cc191 return ArrayDataType::kComplex64; in ConvertDataType()
392 output_array->GetMutableBuffer<ArrayDataType::kComplex64>().data; in ImportComplex64Array()
601 array.data_type = ArrayDataType::kComplex64; in ConvertConstOperator()