| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | conv_weights_converter_test_util.cc | 110 auto weights_shape = OHWI(kDstChannels, 1, 1, kSrcChannels); in ConverterToConvWeights1x1OutX4Test() local
144 auto weights_shape = OHWI(kDstChannels, 1, 1, kSrcChannels); in ConverterToConvWeights1x1OutX4UnalignedTest() local
177 auto weights_shape = OHWI(kDstChannels, 1, 1, kSrcChannels); in ConverterToConvWeights1x1OutX2Test() local
210 auto weights_shape = OHWI(kDstChannels, 3, 4, kSrcChannels); in ConverterToConvWeightsOutX2Test() local
244 auto weights_shape = OHWI(kDstChannels, 4, 4, kSrcChannels); in ConverterToConvTransposedWeights4x4Test() local
279 auto weights_shape = OHWI(dst_channels, 1, 1, src_channels); in ConverterToConvWeights4xTexturesTest() local
|
| D | conv_buffer_1x1.cc | 333 const BHWC& weights_shape, in IsConvBuffer1x1Supported()
404 const Convolution2DAttributes& attr, const BHWC& weights_shape, in CreateConvBuffer1x1DynamicWeights()
|
| D | conv_constants.cc | 98 const OHWI& weights_shape, in GenerateConvolutionConstantCode()
|
| D | conv_powervr.cc | 182 const BHWC& weights_shape, const GpuInfo& gpu_info, in ConvPowerVR()
1393 const Convolution2DAttributes& attr, const BHWC& weights_shape, in GuessBestParams()
1456 const BHWC& weights_shape, in CreateConvPowerVRDynamicWeights()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | weights_broadcast_ops.py | 34 def _has_valid_dims(weights_shape, values_shape): argument
48 weights_rank, weights_shape, values_rank, values_shape): argument
|
| D | nn_batchnorm_test.py | 657 weights_shape, argument
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/sparse_ops/ |
| D | fully_connected.h | 33 const RuntimeShape& weights_shape, const float* weights_data, in FullyConnectedSparseWeight()
79 const RuntimeShape& weights_shape, const float* weights_data, in FullyConnectedSparseWeight1x4Impl()
149 const RuntimeShape& weights_shape; member
167 const RuntimeShape& weights_shape, const float* weights_data, in FullyConnectedSparseWeight1x4()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/default/ |
| D | convolution_selector.cc | 59 const Convolution2DAttributes& attr, const BHWC& weights_shape, in SelectConvolutionDynamicWeightsAdreno()
117 const Convolution2DAttributes& attr, const BHWC& weights_shape, in SelectConvolutionDynamicWeightsMali()
184 const Convolution2DAttributes& attr, const BHWC& weights_shape, in SelectConvolutionWithDynamicWeights()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/sparse_ops/ |
| D | fully_connected.h | 28 const RuntimeShape& weights_shape, const float* weights_data, in FullyConnectedSparseWeight()
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | create_im2col_arrays.cc | 40 const auto& weights_shape = weights_array.shape(); in ProcessConvOperator() local
|
| D | convert_pure_conv_to_depthwise.cc | 98 const auto& weights_shape = weights_array.shape(); in Run() local
|
| D | ensure_bias_vectors.cc | 31 const auto& weights_shape = model.GetArray(weights_name).shape(); in GetOutputDepthFromWeights() local
|
| D | shuffle_fc_weights.cc | 91 const Shape& weights_shape = weights_array.shape(); in Run() local
|
| D | fuse_binary_into_following_affine.cc | 70 const Shape& weights_shape = weights.shape(); in FuseAddOrSubParamsIntoFollowingAffine() local
|
| D | fuse_binary_into_preceding_affine.cc | 126 const Shape& weights_shape = weights.shape(); in FuseMulOrDivParamsIntoPrecedingAffine() local
|
| D | propagate_fixed_sizes.cc | 141 const auto& weights_shape = weights_array.shape(); in ProcessConvOperator() local
203 const auto& weights_shape = weights_array.shape(); in ProcessTransposeConvOperator() local
268 const auto& weights_shape = weights_array.shape(); in ProcessDepthwiseConvOperator() local
383 const auto& weights_shape = weights_array.shape(); in ProcessFullyConnectedOperator() local
1030 const auto& weights_shape = weights_array.shape(); in ProcessLstmCellOperator() local
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/ |
| D | operation_selector.cc | 201 BHWC weights_shape(second_shape.c, 1, 1, second_shape.w); in GPUOperationFromNode() local
288 auto weights_shape = inputs[1]->tensor.shape; in GPUOperationFromNode() local
337 OHWI weights_shape = in GPUOperationFromNode() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | fully_connected.h | 28 const float* input_data, const RuntimeShape& weights_shape, in FullyConnected()
166 const uint8_t* input_data, const RuntimeShape& weights_shape, in ShuffledFullyConnected()
|
| D | reference_ops.h | 442 const float* prev_activ_data, const RuntimeShape& weights_shape, in LstmCell()
646 const RuntimeShape& weights_shape, in LstmCell()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | transpose_conv.cc | 180 const RuntimeShape& weights_shape = GetTensorShape(weights); in ResizeCol2ImTensor() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 644 const float* input_data, const RuntimeShape& weights_shape, in FullyConnected()
1156 const RuntimeShape& weights_shape, in GEMVForLstmCell()
1345 const RuntimeShape& weights_shape, const uint8* weights_data, in GEMVForLstmCellWithSymmetricRange()
2215 const uint8* input_data, const RuntimeShape& weights_shape, in ShuffledFullyConnected()
2982 const float* prev_activ_data, const RuntimeShape& weights_shape, in LstmCell()
3124 const uint8* prev_activ_data_uint8, const RuntimeShape& weights_shape, in LstmCell()
|
| D | optimized_ops.h | 334 const float* input_data, const RuntimeShape& weights_shape, in FullyConnected()
813 const uint8* input_data, const RuntimeShape& weights_shape, in ShuffledFullyConnected()
2777 const float* prev_activ_data, const RuntimeShape& weights_shape, in LstmCell()
2899 const uint8* prev_activ_data_uint8, const RuntimeShape& weights_shape, in LstmCell()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | tooling_util.cc | 2431 const auto& weights_shape = weights_array.shape(); in UndoWeightsShuffling() local
|
| D | export_tensorflow.cc | 1514 const auto& weights_shape = weights_array.shape(); in ConvertLstmCellOperator() local
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | model_builder.cc | 2112 auto weights_shape = graph->FindInputs(node->id)[1]->tensor.shape; in Parse() local
|