| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | resize_nearest_neighbor_test.cc | 32 const RuntimeShape& output_shape, in TestReferenceResizeNearestNeighbor()
51 RuntimeShape output_shape = {1, 1, 1, 1}; in TEST() local
62 RuntimeShape output_shape = {1, 3, 3, 1}; in TEST() local
73 RuntimeShape output_shape = {1, 2, 2, 1}; in TEST() local
84 RuntimeShape output_shape = {1, 2, 5, 1}; in TEST() local
96 RuntimeShape output_shape = {1, 3, 3, 1}; in TEST() local
107 RuntimeShape output_shape = {1, 5, 2, 1}; in TEST() local
118 RuntimeShape output_shape = {1, 4, 4, 1}; in TEST() local
136 RuntimeShape output_shape = {2, 3, 3, 2}; in TEST() local
159 RuntimeShape output_shape({batch, output_height, output_width, depth}); in TestOptimizedResizeNearestNeighbor() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | sparse_to_dense.cc | 44 TfLiteStatus Resize(TfLiteContext* context, const TfLiteTensor* output_shape, in Resize()
57 const TfLiteTensor* output_shape, in CheckDimensionsMatch()
133 const TfLiteTensor* output_shape, in ResizeOutputShape()
151 const TfLiteTensor* output_shape = in Prepare() local
193 const TfLiteTensor* output_shape = in SparseToDenseImpl() local
|
| D | audio_spectrogram_test.cc | 75 std::vector<int> output_shape = m.GetOutputShape(); in TEST() local
91 std::vector<int> output_shape = m.GetOutputShape(); in TEST() local
107 std::vector<int> output_shape = m.GetOutputShape(); in TEST() local
|
| D | reshape.cc | 32 TfLiteIntArray* output_shape) { in ResizeOutput()
66 TfLiteIntArray* output_shape = TfLiteIntArrayCreate(shape->dims->data[0]); in GetOutputShapeFromTensor() local
87 TfLiteIntArray* output_shape = TfLiteIntArrayCreate(num_dimensions); in GetOutputShapeFromParam() local
|
| D | transpose_conv.cc | 107 const TfLiteTensor* output_shape, in ResizeIm2ColTensor()
140 const TfLiteTensor* output_shape = in Prepare() local
176 const TfLiteTensor* output_shape = in Eval() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reference_ops.h | 136 const float* bias_data, const RuntimeShape& output_shape, in Conv()
209 const int32* bias_data, const RuntimeShape& output_shape, in Conv()
299 const RuntimeShape output_shape = in DepthToSpace() local
352 const RuntimeShape output_shape = in SpaceToDepth() local
395 const RuntimeShape& output_shape, float* output_data) { in Elu()
404 const RuntimeShape& output_shape, float* output_data) { in Relu()
415 const RuntimeShape& output_shape, float* output_data) { in Relu1()
428 const RuntimeShape& output_shape, float* output_data) { in Relu6()
443 const RuntimeShape& output_shape, T* output_data) { in ReluX()
458 const RuntimeShape& output_shape, float* output_data) { in LeakyRelu()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_layout_assignment.cc | 107 Shape* output_shape; in AddBackendConstraintsToDnnConvCustomCall() local
209 Shape output_shape = instruction->shape(); in AddBackendConstraints() local
259 Shape output_shape = instruction->shape(); in AddBackendConstraints() local
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | cudnn_rnn_ops_test.cc | 50 std::vector<int> output_shape = {seq_length, batch_size, in TEST() local
82 std::vector<int> output_shape = {seq_length, batch_size, in TEST() local
114 std::vector<int> output_shape = {max_seq_length, batch_size, in TEST() local
|
| D | sdca_ops.cc | 109 ShapeHandle output_shape; in __anonac679a370102() local
|
| D | cudnn_rnn_ops.cc | 95 auto output_shape = c->MakeShape({seq_length, batch_size, output_size}); in __anone1d60d010302() local
138 auto output_shape = c->MakeShape({seq_length, batch_size, output_size}); in __anone1d60d010402() local
184 auto output_shape = in __anone1d60d010502() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/ |
| D | mul.h | 45 const RuntimeShape& output_shape, int8_t* output_data) { in Mul()
59 const RuntimeShape& output_shape, int8_t* output_data) { in Mul()
90 const RuntimeShape& output_shape, in BroadcastMul4DSlow()
|
| /external/tensorflow/tensorflow/contrib/distributions/python/ops/bijectors/ |
| D | softmax_centered.py | 90 def _inverse_event_shape(self, output_shape): argument
97 def _inverse_event_shape_tensor(self, output_shape): argument
|
| D | ordered.py | 83 def _inverse_event_shape(self, output_shape): argument
90 def _inverse_event_shape_tensor(self, output_shape): argument
|
| D | invert.py | 92 def _inverse_event_shape(self, output_shape): argument
95 def _inverse_event_shape_tensor(self, output_shape): argument
|
| D | inline.py | 123 def _inverse_event_shape(self, output_shape): argument
129 def _inverse_event_shape_tensor(self, output_shape): argument
|
| D | reshape.py | 327 def _inverse_event_shape(self, output_shape): argument
339 def _inverse_event_shape_tensor(self, output_shape): argument
|
| D | chain.py | 242 def _inverse_event_shape(self, output_shape): argument
246 def _inverse_event_shape_tensor(self, output_shape): argument
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 299 int accum_shift, const RuntimeShape& output_shape, in GEMVForLstmCell()
486 int32 accum_multiplier, int accum_shift, const RuntimeShape& output_shape, in GEMVForLstmCellWithSymmetricRange()
777 const float* optional_bias_data, const RuntimeShape& output_shape, in FullyConnected()
822 int32 output_activation_max, const RuntimeShape& output_shape, in FullyConnectedAsGEMVWorkerImpl()
1098 int32 output_activation_max, const RuntimeShape& output_shape, in FullyConnectedAsGEMV()
1170 const int32* bias_data, const RuntimeShape& output_shape, in FullyConnected()
1229 const int32* bias_data_int32, const RuntimeShape& output_shape, in FullyConnected()
1635 const int32* bias_data, const RuntimeShape& output_shape, in ShuffledFullyConnected()
1759 const RuntimeShape& output_shape, uint8_t* output_data, in MeanImpl()
1931 const RuntimeShape output_shape = in Mean() local
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | deserialize_sparse_string_op.cc | 82 Tensor output_shape; in Compute() local
105 Tensor output_shape; in Compute() local
229 Tensor* output_indices, Tensor* output_values, Tensor* output_shape) { in GetAndValidateSparseTensor()
|
| D | deserialize_sparse_variant_op.cc | 61 const Tensor* output_shape; in Compute() local
91 Tensor* output_shape; in Compute() local
278 int index, const Tensor** output_shape, in GetAndValidateSparseTensorShape()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | broadcast_to_op.cc | 32 TensorShape output_shape; in Compile() local
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | convert_squeeze_to_reshape.cc | 61 const auto& output_shape = model->GetArray(squeeze_op->outputs[0]).shape(); in Run() local
|
| D | propagate_fixed_sizes.cc | 37 PaddingType padding_type, Shape* output_shape, in ComputeConvSizes()
156 const auto& output_shape = output_array.shape(); in ProcessConvOperator() local
457 auto& output_shape = *output_array.mutable_shape(); in ProcessTensorFlowReshapeOperator() local
649 Shape* output_shape = model->GetArray(output_name).mutable_shape(); in ProcessReorderAxesOperator() local
689 auto& output_shape = *output_array.mutable_shape(); in ProcessConcatenationOperator() local
741 Shape* output_shape = output_array.mutable_shape(); in ProcessRangeOperator() local
781 Shape output_shape = input_shape; in ProcessTensorFlowSplitOperator() local
875 Shape output_shape = input_shape; in ProcessTensorFlowSplitVOperator() local
954 std::vector<int32> output_shape = in ProcessResizeBilinearOperator() local
982 std::vector<int32> output_shape = in ProcessResizeNearestNeighborOperator() local
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dynamic_dimension_inference_test.cc | 339 auto output_shape = ShapeUtil::MakeShape(F32, {3, 2, 1}); in TEST_F() local
378 auto output_shape = ShapeUtil::MakeShape(F32, {6, 4, 1, 5, 2, 3}); in TEST_F() local
412 auto output_shape = ShapeUtil::MakeShape(F32, {6, 4, 1, 5, 2, 3}); in TEST_F() local
437 auto output_shape = ShapeUtil::MakeShape(F32, {3, 2, 4}); in TEST_F() local
464 auto output_shape = ShapeUtil::MakeShape(F32, {2, 2, 2}); in TEST_F() local
554 auto output_shape = ShapeUtil::MakeShape(F32, {2, 2, 2}); in TEST_F() local
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 421 int accum_shift, const RuntimeShape& output_shape, in GEMVForLstmCell()
608 int32 accum_multiplier, int accum_shift, const RuntimeShape& output_shape, in GEMVForLstmCellWithSymmetricRange()
899 const float* bias_data, const RuntimeShape& output_shape, in FullyConnected()
967 int32 output_activation_max, const RuntimeShape& output_shape, in FullyConnectedAsGEMV()
1153 const int32* bias_data, const RuntimeShape& output_shape, in FullyConnected()
1237 const int32* bias_data_int32, const RuntimeShape& output_shape, in FullyConnected()
1691 const int32* bias_data, const RuntimeShape& output_shape, in ShuffledFullyConnected()
1943 const RuntimeShape& output_shape, T* im2col_data) { in DilatedIm2col()
2049 const RuntimeShape& output_shape, T* output_data) { in Im2col()
2114 const float* bias_data, const RuntimeShape& output_shape, in Conv()
[all …]
|