Home
last modified time | relevance | path

Searched defs:weights_type (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dconvolution_transposed_3x3_thin.cc208 DataType weights_type = definition_.precision == CalculationsPrecision::F32 in UploadWeights() local
257 const DataType weights_type = definition.GetDataType(); in CreateConvolutionTransposed3x3ThinDynamicWeights() local
Dconvolution_transposed_4x4.cc397 DataType weights_type = definition_.precision == CalculationsPrecision::F32 in UploadWeights() local
450 const DataType weights_type = definition.GetDataType(); in CreateConvolutionTransposed4x4DynamicWeights() local
Dconvolution_transposed_3x3.cc392 DataType weights_type = definition_.precision == CalculationsPrecision::F32 in UploadWeights() local
444 const DataType weights_type = definition.GetDataType(); in CreateConvolutionTransposed3x3DynamicWeights() local
Dconvolution_transposed.h102 DataType weights_type = definition_.precision == CalculationsPrecision::F32 in UploadWeights() local
Dconv_metal.cc594 const WeightsDescription& weights_desc, const DataType& weights_type) { in ReorderWeightsForConv()
1022 auto weights_type = DeduceDataTypeFromPrecision(definition.precision); in CreateConvolutionMetal() local
1147 auto weights_type = DeduceDataTypeFromPrecision(definition.precision); in CreateConvolutionMetalWino4x4To6x6() local
Dconv_powervr.h258 DataType weights_type = conv_params_.weights_data_type; in UploadWeights() local
Dconv_weights_converter_test_util.cc53 DataType weights_type = DataType::FLOAT32; in ConvolutionWeightsConverterTest() local
Dconvolution_transposed.cc620 const DataType weights_type = definition.GetDataType(); in CreateConvolutionTransposedDynamicWeights() local
/external/tensorflow/tensorflow/lite/kernels/
Dtranspose_conv.cc125 TfLiteType weights_type, in AllocateTemporaryTensorsIfRequired()
Dunidirectional_sequence_lstm_test.cc38 const TensorType& weights_type = TensorType_FLOAT32, in UnidirectionalLSTMOpModel()
2474 const TensorType& weights_type = TensorType_FLOAT32) in LayerNormUnidirectionalLSTMOpModel()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate.cc352 const TfLiteType weights_type = context->tensors[weights_id].type; in IsHybridOperator() local
361 const TfLiteType weights_type = context->tensors[weights_id].type; in IsHybridOperator() local
369 const TfLiteType weights_type = context->tensors[weights_id].type; in IsHybridOperator() local
376 const TfLiteType weights_type = context->tensors[weights_id].type; in IsHybridOperator() local
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_ops.cc859 const auto weights_type = weights_tensor.getType().cast<ShapedType>(); in fold() local