| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | directed_interleave_dataset_test.py | 40 def _get_weights_of_type(weights_list, weights_type): argument
80 def testSampleFromDatasets(self, weights_type): argument
111 def testSampleFromDatasetsStoppingOnEmptyDataset(self, weights_type): argument
128 def testSampleFromDatasetsSkippingEmptyDataset(self, weights_type): argument
146 def testSampleFromDatasetsWithZeroWeight(self, weights_type): argument
160 def testSampleFromEmptyDataset(self, weights_type): argument
|
| /external/XNNPACK/test/ |
| D | prelu-operator-tester.h | 80 inline PReLUOperatorTester& weights_type(WeightsType weights_type) { in weights_type() argument
85 inline WeightsType weights_type() const { in weights_type() function
|
| D | fully-connected-operator-tester.h | 130 inline FullyConnectedOperatorTester& weights_type(WeightsType weights_type) { in weights_type() function
135 inline WeightsType weights_type() const { in weights_type() function
|
| D | convolution-operator-tester.h | 505 inline ConvolutionOperatorTester& weights_type(WeightsType weights_type) { in weights_type() function
510 inline WeightsType weights_type() const { in weights_type() function
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | convolution_transposed.h | 102 DataType weights_type = definition_.precision == CalculationsPrecision::F32 in UploadWeights() local
145 DataType weights_type = definition_.precision == CalculationsPrecision::F32 in UploadWeights() local
|
| D | convolution_transposed_3x3_thin.cc | 208 DataType weights_type = definition_.precision == CalculationsPrecision::F32 in UploadWeights() local
257 const DataType weights_type = definition.GetDataType(); in CreateConvolutionTransposed3x3ThinDynamicWeights() local
|
| D | convolution_transposed_4x4.cc | 397 DataType weights_type = definition_.precision == CalculationsPrecision::F32 in UploadWeights() local
450 const DataType weights_type = definition.GetDataType(); in CreateConvolutionTransposed4x4DynamicWeights() local
|
| D | convolution_transposed_3x3.cc | 392 DataType weights_type = definition_.precision == CalculationsPrecision::F32 in UploadWeights() local
444 const DataType weights_type = definition.GetDataType(); in CreateConvolutionTransposed3x3DynamicWeights() local
|
| D | conv_powervr.h | 258 DataType weights_type = conv_params_.weights_data_type; in UploadWeights() local
302 DataType weights_type = conv_params_.weights_data_type; in UploadWeights() local
|
| D | conv_metal.cc | 594 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
|
| D | conv_weights_converter_test_util.cc | 53 DataType weights_type = DataType::FLOAT32; in ConvolutionWeightsConverterTest() local
|
| D | convolution_transposed.cc | 635 const DataType weights_type = definition.GetDataType(); in CreateConvolutionTransposedDynamicWeights() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | transpose_conv.cc | 125 TfLiteType weights_type, in AllocateTemporaryTensorsIfRequired()
|
| D | unidirectional_sequence_lstm_test.cc | 38 const TensorType& weights_type = TensorType_FLOAT32, in UnidirectionalLSTMOpModel()
2474 const TensorType& weights_type = TensorType_FLOAT32) in LayerNormUnidirectionalLSTMOpModel()
|
| /external/ComputeLibrary/src/runtime/NEON/functions/assembly/ |
| D | NEDepthwiseConvolutionAssemblyDispatch.cpp | 478 const DataType weights_type = weights->data_type(); in is_optimized_supported() local
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate.cc | 375 const TfLiteType weights_type = context->tensors[weights_id].type; in IsHybridOperator() local
384 const TfLiteType weights_type = context->tensors[weights_id].type; in IsHybridOperator() local
392 const TfLiteType weights_type = context->tensors[weights_id].type; in IsHybridOperator() local
399 const TfLiteType weights_type = context->tensors[weights_id].type; in IsHybridOperator() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.cc | 946 const auto weights_type = weights_tensor.getType().cast<ShapedType>(); in fold() local
|