| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | directed_interleave_dataset_test.py | 35 def _get_weights_of_type(weights_list, weights_type): argument
75 def testSampleFromDatasets(self, weights_type): argument
106 def testSampleFromDatasetsStoppingOnEmptyDataset(self, weights_type): argument
123 def testSampleFromDatasetsSkippingEmptyDataset(self, weights_type): argument
141 def testSampleFromDatasetsWithZeroWeight(self, weights_type): argument
155 def testSampleFromEmptyDataset(self, weights_type): argument
|
| /external/XNNPACK/test/ |
| D | prelu-operator-tester.h | 81 inline PReLUOperatorTester& weights_type(WeightsType weights_type) { in weights_type() function
86 inline WeightsType weights_type() const { in weights_type() function
|
| D | fully-connected-operator-tester.h | 131 inline FullyConnectedOperatorTester& weights_type(WeightsType weights_type) { in weights_type() function
136 inline WeightsType weights_type() const { in weights_type() function
|
| D | deconvolution-operator-tester.h | 421 inline DeconvolutionOperatorTester& weights_type(WeightsType weights_type) { in weights_type() argument
426 inline WeightsType weights_type() const { in weights_type() function
|
| D | convolution-operator-tester.h | 507 inline ConvolutionOperatorTester& weights_type(WeightsType weights_type) { in weights_type() function
512 inline WeightsType weights_type() const { in weights_type() function
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | conv_metal_simd.cc | 378 const DataType& weights_type, int dst_x8_slices) { in ReorderWeightsForConv()
481 auto weights_type = DeduceDataTypeFromPrecision(definition.precision); in CreateConvolutionMetalSimd() local
|
| D | convolution_transposed_3x3_thin.cc | 266 const DataType weights_type = definition.GetDataType(); in CreateConvolutionTransposed3x3ThinDynamicWeights() local
|
| D | convolution_transposed_4x4.cc | 422 const DataType weights_type = definition.GetDataType(); in CreateConvolutionTransposed4x4DynamicWeights() local
|
| D | convolution_transposed_3x3.cc | 432 const DataType weights_type = definition.GetDataType(); in CreateConvolutionTransposed3x3DynamicWeights() local
|
| D | conv_weights_converter_test_util.cc | 65 DataType weights_type = DataType::FLOAT32; in ConvolutionWeightsConverterTest() local
|
| D | convolution_transposed.cc | 641 const DataType weights_type = definition.GetDataType(); in CreateConvolutionTransposedDynamicWeights() local
|
| D | conv_generic.cc | 270 const DataType weights_type = definition_.GetDataType(); in GenerateCode() local
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| D | conv_2d_helpers.cc | 154 void Conv2dOpBuilder::CheckShouldSplitDwConv(TfLiteType weights_type, in CheckShouldSplitDwConv()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | transpose_conv.cc | 125 TfLiteType weights_type, in AllocateTemporaryTensorsIfRequired()
|
| D | unidirectional_sequence_lstm_test.cc | 2222 const TensorType& weights_type = TensorType_FLOAT32) in LayerNormUnidirectionalLSTMOpModel()
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate.cc | 401 const TfLiteType weights_type = context->tensors[weights_id].type; in IsHybridOperator() local
410 const TfLiteType weights_type = context->tensors[weights_id].type; in IsHybridOperator() local
418 const TfLiteType weights_type = context->tensors[weights_id].type; in IsHybridOperator() local
425 const TfLiteType weights_type = context->tensors[weights_id].type; in IsHybridOperator() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.cc | 1088 const auto weights_type = weights_tensor.getType().cast<ShapedType>(); in fold() local
|