| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | convert_tensor_test.cc | 33 PartialTensorShape output_shape = in TEST() local
42 PartialTensorShape output_shape = ConvertTypeToTensorShape( in TEST() local
51 PartialTensorShape output_shape = ConvertTypeToTensorShape( in TEST() local
60 PartialTensorShape output_shape = ConvertTypeToTensorShape(b.getF32Type()); in TEST() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | transpose_utils_test.cc | 25 RuntimeShape output_shape({9}); in TEST() local
43 RuntimeShape output_shape({3, 9}); in TEST() local
63 RuntimeShape output_shape({1, 9}); in TEST() local
82 RuntimeShape output_shape({8, 4, 3}); in TEST() local
104 RuntimeShape output_shape({8, 4, 1}); in TEST() local
126 RuntimeShape output_shape({1, 4, 1}); in TEST() local
146 RuntimeShape output_shape({1, 1, 1}); in TEST() local
166 RuntimeShape output_shape({3, 9, 4, 2}); in TEST() local
190 RuntimeShape output_shape({3, 9, 4, 1}); in TEST() local
213 RuntimeShape output_shape({3, 1, 4, 1}); in TEST() local
[all …]
|
| 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/internal/reference/ |
| D | pad.h | 40 const P* pad_value_ptr, const RuntimeShape& output_shape, in PadImpl()
113 const P* pad_value_ptr, const RuntimeShape& output_shape, in Pad()
123 const int32* pad_value_ptr, const RuntimeShape& output_shape, in Pad()
134 const int32* pad_value_ptr, const RuntimeShape& output_shape, in Pad()
147 const RuntimeShape& output_shape, T* output_data) { in PadImageStyle()
155 const RuntimeShape& output_shape, in PadImageStyle()
165 const RuntimeShape& output_shape, in PadImageStyle()
175 const RuntimeShape& output_shape, in PadImageStyle()
|
| D | logistic.h | 31 const RuntimeShape& output_shape, float* output_data) { in Logistic()
62 const float* input_data, const RuntimeShape& output_shape, in Logistic()
70 const RuntimeShape& output_shape, int16* output_data) { in Logistic()
|
| D | reference_ops.h | 77 const RuntimeShape output_shape = in DepthToSpace() local
130 const RuntimeShape output_shape = in SpaceToDepth() local
173 const RuntimeShape& output_shape, float* output_data) { in Elu()
183 const RuntimeShape& output_shape, T* output_data) { in Relu()
195 const RuntimeShape& output_shape, T* output_data) { in Relu1()
208 const RuntimeShape& output_shape, float* output_data) { in Relu6()
223 const RuntimeShape& output_shape, T* output_data) { in ReluX()
241 const RuntimeShape& output_shape, T* output_data) { in ReluX()
256 const RuntimeShape& output_shape, float* output_data) { in LeakyRelu()
270 const RuntimeShape& output_shape, in QuantizeLeakyRelu()
[all …]
|
| D | comparisons.h | 59 const T* input2_data, const RuntimeShape& output_shape, bool* output_data) { in ComparisonImpl()
73 const RuntimeShape& output_shape, bool* output_data) { in Comparison()
82 const T* input2_data, const RuntimeShape& output_shape, bool* output_data) { in ComparisonWithScaling()
117 const RuntimeShape output_shape = in BroadcastComparison4DSlowImpl() local
143 const RuntimeShape& output_shape, in BroadcastComparison4DSlow()
159 const RuntimeShape output_shape = in BroadcastComparison4DSlowWithScaling() local
|
| D | mul.h | 48 const RuntimeShape& output_shape, T* output_data) { in Mul()
65 const RuntimeShape& output_shape, uint8* output_data) { in Mul()
79 const RuntimeShape& output_shape, in BroadcastMul4DSlow()
129 const RuntimeShape output_shape = in BroadcastMul4DSlow() local
|
| D | add.h | 29 const RuntimeShape& output_shape, T* output_data) { in Add()
42 const RuntimeShape& output_shape, float* output_data) { in Add()
122 const RuntimeShape& output_shape, uint8* output_data) { in Add()
138 const RuntimeShape& output_shape, int16* output_data) { in Add()
184 const RuntimeShape& output_shape, in BroadcastAdd4DSlow()
224 const RuntimeShape& output_shape, in BroadcastAdd4DSlow()
265 const RuntimeShape& output_shape, in BroadcastAdd4DSlow()
328 const RuntimeShape& output_shape, in BroadcastAddFivefold()
|
| D | dequantize.h | 32 const RuntimeShape& output_shape, OutputT* output_data) { in Dequantize()
49 const RuntimeShape& output_shape, in DequantizeInteger()
|
| D | binary_function.h | 41 const RuntimeShape output_shape = in BroadcastBinaryFunction4DSlow() local
72 const RuntimeShape& output_shape, R* output_data, in BinaryFunction()
|
| /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
199 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 | 37 TfLiteIntArray* output_shape = GetOutputShape(context, node); in ResizeOutput() local
74 TfLiteIntArray* output_shape = TfLiteIntArrayCreate(shape->dims->data[0]); in GetOutputShapeFromTensor() local
95 TfLiteIntArray* output_shape = TfLiteIntArrayCreate(num_dimensions); in GetOutputShapeFromParam() local
|
| D | mfcc_test.cc | 84 std::vector<int> output_shape = m.GetOutputShape(); in TEST() local
109 std::vector<int> output_shape = m.GetOutputShape(); in TEST() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_layout_assignment.cc | 123 Shape* output_shape; in AddBackendConstraintsToDnnConvCustomCall() local
226 Shape output_shape = instruction->shape(); in AddBackendConstraints() local
238 Shape output_shape = instruction->shape(); in AddBackendConstraints() local
286 Shape output_shape = instruction->shape(); in AddBackendConstraints() local
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | activations.cc | 35 const Q* input_data, const RuntimeShape& output_shape, in ReluQuantized()
46 const RuntimeShape& output_shape, float* output_data) { in ReluFloat()
57 const RuntimeShape& output_shape, float* output_data) { in Relu6Float()
71 const RuntimeShape& output_shape, Q* output_data) { in Relu6Quantized()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | segment_reduction_ops_impl.h | 113 TensorShape output_shape = input.shape(); in Compute() local
246 TensorShape output_shape = input.shape(); in ComputeAsync() local
286 TensorShape output_shape = input.shape(); in ComputeAsync() local
|
| 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/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
116 std::vector<int> output_shape = {max_seq_length, batch_size, in TEST() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/ |
| D | mul.h | 46 const RuntimeShape& output_shape, int8_t* output_data) { in Mul()
60 const RuntimeShape& output_shape, int8_t* output_data) { in Mul()
91 const RuntimeShape& output_shape, in BroadcastMul4DSlow()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| D | lstm_utils_test.cc | 49 SmallVector<int64_t, 2> output_shape{1, 2}; in createLstmCompositeFunc() local
179 SmallVector<int64_t, 2> output_shape{1, -1}; in TEST_F() local
248 SmallVector<int64_t, 2> output_shape{1, -1}; in TEST_F() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 278 const float* optional_bias_data, const RuntimeShape& output_shape, in FullyConnected()
312 const int32* bias_data, const RuntimeShape& output_shape, in FullyConnected()
373 const int32* bias_data_int32, const RuntimeShape& output_shape, in FullyConnected()
751 const int32* bias_data, const RuntimeShape& output_shape, in ShuffledFullyConnected()
924 const RuntimeShape& output_shape, uint8_t* output_data, in MeanImpl()
1072 const RuntimeShape& output_shape; member
1091 const RuntimeShape output_shape = in Mean() local
1149 const float* bias_data, const RuntimeShape& output_shape, in Conv()
1250 int32_t* accum_scratch, const RuntimeShape& output_shape, in HybridConv()
1336 const RuntimeShape& output_shape, float* output_data, in HybridConvPerChannel()
[all …]
|
| /external/tensorflow/tensorflow/lite/experimental/delegates/hexagon/builders/ |
| D | reshape_builder.cc | 31 std::vector<int>* output_shape) { in PopulateOutputShapeFromTensor()
38 std::vector<int>* output_shape) { in PopulateShapeFromParam()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/ |
| D | pad_test.cc | 36 const BHWC& output_shape, std::vector<float>&& expected) { in TestPadOperation()
58 void TestPrepending(const HWC& prepend, const BHWC& output_shape, in TestPrepending()
63 void TestAppending(const HWC& append, const BHWC& output_shape, in TestAppending()
|