Home
last modified time | relevance | path

Searched refs:kComplex64 (Results 1 – 19 of 19) 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_gather.cc135 case ArrayDataType::kComplex64: in Run()
136 Gather<ArrayDataType::kComplex64>(input_array, coords_array, 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_slice.cc150 case ArrayDataType::kComplex64: in Run()
151 if (!Slice<ArrayDataType::kComplex64>(*op, input_array, &output_array)) { in Run()
Dresolve_constant_tile.cc154 case ArrayDataType::kComplex64: in Run()
155 Tile<ArrayDataType::kComplex64>(input_array, multiples_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.cc105 case ArrayDataType::kComplex64: in Serialize()
133 return ArrayDataType::kComplex64; in Deserialize()
160 case ArrayDataType::kComplex64: in Serialize()
161 return CopyBuffer<ArrayDataType::kComplex64>(array, builder); in Serialize()
190 return CopyBuffer<ArrayDataType::kComplex64>(buffer, array); in Deserialize()
Dtypes_test.cc78 {ArrayDataType::kComplex64, ::tflite::TensorType_COMPLEX64}}; in TEST()
185 Array recovered = ToFlatBufferAndBack<ArrayDataType::kComplex64>( in TEST()
187 EXPECT_THAT(recovered.GetBuffer<ArrayDataType::kComplex64>().data, in TEST()
Doperator.cc55 {ArrayDataType::kComplex64, ::tflite::TensorType_COMPLEX64}, in GetTensorType()
/external/tensorflow/tensorflow/lite/toco/
Dtooling_util.cc90 case ArrayDataType::kComplex64: in ArrayDataTypeName()
850 case ArrayDataType::kComplex64: in CompareConstantArrays()
851 return CompareArrayBuffers<ArrayDataType::kComplex64>(lhs_array, in CompareConstantArrays()
1454 case ArrayDataType::kComplex64: in CloneArray()
1455 CopyArrayBuffer<ArrayDataType::kComplex64>(source_array, &target_array); in CloneArray()
1770 case ArrayDataType::kComplex64: in ElementSize()
2321 return ArrayDataType::kComplex64; in ConvertIODataTypeToArrayDataType()
Dmodel.h236 kComplex64, enumerator
294 struct DataTypeImpl<ArrayDataType::kComplex64> {
Dimport_tensorflow_test.cc399 constexpr ArrayDataType kType = ArrayDataType::kComplex64; in TEST_F()
Dexport_tensorflow.cc69 case ArrayDataType::kComplex64: in GetTensorFlowDataType()
306 const auto& data = array.GetBuffer<ArrayDataType::kComplex64>().data; in ConvertComplex64TensorConst()
2453 case ArrayDataType::kComplex64: in AddPlaceholder()
2515 case ArrayDataType::kComplex64: in ExportTensorFlowGraphDefImplementation()
Dimport_tensorflow.cc196 return ArrayDataType::kComplex64; in ConvertDataType()
409 output_array->GetMutableBuffer<ArrayDataType::kComplex64>().data; in ImportComplex64Array()
813 array.data_type = ArrayDataType::kComplex64; in ConvertConstOperator()