Home
last modified time | relevance | path

Searched defs:output_shape (Results 1 – 25 of 433) sorted by relevance

12345678910>>...18

/external/tensorflow/tensorflow/lite/kernels/internal/
Dresize_nearest_neighbor_test.cc32 const RuntimeShape& output_shape, in TestReferenceResizeNearestNeighbor()
52 RuntimeShape output_shape = {1, 1, 1, 1}; in TEST() local
63 RuntimeShape output_shape = {1, 1, 1, 1}; in TEST() local
75 RuntimeShape output_shape = {1, 1, 1, 1}; in TEST() local
87 RuntimeShape output_shape = {1, 3, 3, 1}; in TEST() local
98 RuntimeShape output_shape = {1, 3, 3, 1}; in TEST() local
109 RuntimeShape output_shape = {1, 3, 3, 1}; in TEST() local
121 RuntimeShape output_shape = {1, 3, 3, 1}; in TEST() local
133 RuntimeShape output_shape = {1, 2, 2, 1}; in TEST() local
144 RuntimeShape output_shape = {1, 2, 2, 1}; in TEST() local
[all …]
Dtranspose_utils_test.cc25 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 …]
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dreshape_test.cc40 std::vector<int32_t> output_shape(input_shape.cbegin(), input_shape.cend()); variable
61 std::vector<int32_t> output_shape(input_shape.cbegin(), input_shape.cend()); variable
82 std::vector<int32_t> output_shape(input_shape.cbegin(), input_shape.cend()); variable
103 std::vector<int32_t> output_shape(input_shape.cbegin(), input_shape.cend()); variable
123 std::vector<int32_t> output_shape(input_shape.cbegin(), input_shape.cend()); variable
143 std::vector<int32_t> output_shape(input_shape.cbegin(), input_shape.cend()); variable
214 std::vector<int32_t> output_shape(input_shape.cbegin(), input_shape.cend()); in TEST() local
/external/tensorflow/tensorflow/lite/micro/kernels/
Dstrided_slice_test.cc64 const int* output_shape, float* output_data, in TestStridedSliceFloat()
94 const int32_t* strides_data, const int* output_shape, T* output_data, in TestStridedSliceQuantized()
131 const int output_shape[] = {0}; in TF_LITE_MICRO_TEST() local
152 const int output_shape[] = {1, 2}; in TF_LITE_MICRO_TEST() local
173 const int output_shape[] = {1, 0}; in TF_LITE_MICRO_TEST() local
194 const int output_shape[] = {1, 2}; in TF_LITE_MICRO_TEST() local
215 const int output_shape[] = {1, 3}; in TF_LITE_MICRO_TEST() local
236 const int output_shape[] = {1, 1}; in TF_LITE_MICRO_TEST() local
257 const int output_shape[] = {1, 3}; in TF_LITE_MICRO_TEST() local
278 const int output_shape[] = {1, 3}; in TF_LITE_MICRO_TEST() local
[all …]
Dpooling_test.cc192 const int output_shape[] = {4, 1, 1, 2, 1}; in TF_LITE_MICRO_TEST() local
208 const int output_shape[] = {4, 1, 1, 2, 1}; in TF_LITE_MICRO_TEST() local
230 const int output_shape[] = {4, 1, 1, 2, 1}; in TF_LITE_MICRO_TEST() local
252 const int output_shape[] = {4, 1, 1, 3, 1}; in TF_LITE_MICRO_TEST() local
275 const int output_shape[] = {4, 1, 1, 2, 1}; in TF_LITE_MICRO_TEST() local
297 const int output_shape[] = {4, 1, 1, 2, 1}; in TF_LITE_MICRO_TEST() local
319 const int output_shape[] = {4, 1, 2, 4, 1}; in TF_LITE_MICRO_TEST() local
341 const int output_shape[] = {4, 1, 1, 2, 1}; in TF_LITE_MICRO_TEST() local
357 const int output_shape[] = {4, 1, 1, 2, 1}; in TF_LITE_MICRO_TEST() local
373 const int output_shape[] = {4, 1, 1, 2, 1}; in TF_LITE_MICRO_TEST() local
[all …]
Dcomparisons.cc50 RuntimeShape output_shape = tflite::micro::GetTensorShape(output); in EqualEval() local
155 RuntimeShape output_shape = tflite::micro::GetTensorShape(output); in NotEqualEval() local
259 RuntimeShape output_shape = tflite::micro::GetTensorShape(output); in GreaterEval() local
350 RuntimeShape output_shape = tflite::micro::GetTensorShape(output); in GreaterEqualEval() local
441 RuntimeShape output_shape = tflite::micro::GetTensorShape(output); in LessEval() local
532 RuntimeShape output_shape = tflite::micro::GetTensorShape(output); in LessEqualEval() local
Dactivations.cc52 const RuntimeShape& output_shape, const T* input_data, in ReluQuantized()
77 const RuntimeShape output_shape = GetTensorShape(output); in CalculateReluOpData() local
98 const RuntimeShape& output_shape, float* output_data) { in ReluFloat()
109 const RuntimeShape& output_shape, float* output_data) { in Relu6Float()
123 const RuntimeShape& output_shape, Q* output_data) { in Relu6Quantized()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_sharding_util_test.cc38 Shape output_shape = ShapeUtil::MakeShape(F32, {3, 5, 2}); in TEST() local
48 Shape output_shape = ShapeUtil::MakeShape(F32, {3, 5, 2}); in TEST() local
57 Shape output_shape = ShapeUtil::MakeShape(F32, {20, 7}); in TEST() local
69 Shape output_shape = ShapeUtil::MakeShape(F32, {4, 4, 7}); in TEST() local
81 Shape output_shape = ShapeUtil::MakeShape(F32, {4, 4, 7}); in TEST() local
95 Shape output_shape = ShapeUtil::MakeShape(F32, {4, 16, 7}); in TEST() local
108 Shape output_shape = ShapeUtil::MakeShape(F32, {4, 15, 2, 14}); in TEST() local
121 Shape output_shape = ShapeUtil::MakeShape(F32, {3, 5, 1, 7}); in TEST() local
134 Shape output_shape = ShapeUtil::MakeShape(F32, {8, 16, 1}); in TEST() local
154 Shape output_shape = ShapeUtil::MakeShape(F32, {}); in TEST() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dpad.h40 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_t* pad_value_ptr, const RuntimeShape& output_shape, in Pad()
134 const int32_t* pad_value_ptr, const RuntimeShape& output_shape, in Pad()
144 const RuntimeShape& output_shape, T* output_data) { in PadImageStyle()
153 const RuntimeShape& output_shape, in PadImageStyle()
Dsub.h37 const RuntimeShape& output_shape, in SubNonBroadcast()
53 const RuntimeShape& output_shape, in SubNonBroadcast()
74 const RuntimeShape& output_shape, in BroadcastSubSlow()
114 const RuntimeShape& output_shape, in BroadcastSubSlow()
171 const RuntimeShape& output_shape, in BroadcastSubSlow()
211 const RuntimeShape& output_shape, in BroadcastSubSlow()
265 const RuntimeShape& output_shape, int64_t* output_data) { in BroadcastSubSlow()
302 const RuntimeShape& output_shape, T* output_data) { in BroadcastSubSlow()
341 const RuntimeShape& output_shape, in BroadcastSub16POTSlow()
449 const RuntimeShape& output_shape, uint8_t* output_data) { in Sub()
[all …]
Dlogistic.h31 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_t* output_data) { in Logistic()
95 const RuntimeShape& output_shape, int8_t* output_data, in Logistic()
Dreference_ops.h92 const RuntimeShape& output_shape, T* output_data) { in Relu()
104 const RuntimeShape& output_shape, T* output_data) { in Relu1()
117 const RuntimeShape& output_shape, float* output_data) { in Relu6()
132 const RuntimeShape& output_shape, T* output_data) { in ReluX()
150 const RuntimeShape& output_shape, T* output_data) { in ReluX()
173 const RuntimeShape& output_shape, in BroadcastMulFivefold()
223 const RuntimeShape& output_shape, int16* output_data) { in Mul()
242 const RuntimeShape& output_shape, uint8* output_data) { in Mul()
271 const RuntimeShape& output_shape, int16_t* output_data) { in Sub16()
318 const Scalar* const* input_data, const RuntimeShape& output_shape, in Pack()
[all …]
Dtanh.h30 const RuntimeShape& output_shape, float* output_data) { in Tanh()
43 const float* input_data, const RuntimeShape& output_shape, in Tanh()
50 const int16_t* input_data, const RuntimeShape& output_shape, in Tanh()
84 const uint8_t* input_data, const RuntimeShape& output_shape, in Tanh()
Dadd.h29 const RuntimeShape& output_shape, T* output_data) { in Add()
42 const RuntimeShape& output_shape, float* output_data) { in Add()
127 const RuntimeShape& output_shape, uint8_t* output_data) { in Add()
145 const RuntimeShape& output_shape, in AddGeneralParamScale()
210 const RuntimeShape& output_shape, in BroadcastAdd4DSlow()
250 const RuntimeShape& output_shape, in BroadcastAdd4DSlow()
293 const T* input2_data, const RuntimeShape& output_shape, T* output_data) { in BroadcastAdd4DSlow()
355 const RuntimeShape& output_shape, in BroadcastAddFivefold()
Dmul.h48 const RuntimeShape& output_shape, T* output_data) { in Mul()
65 const RuntimeShape& output_shape, uint8_t* output_data) { in Mul()
79 const RuntimeShape& output_shape, in BroadcastMul4DSlow()
129 const RuntimeShape output_shape = in BroadcastMul4DSlow() local
Dleaky_relu.h28 const RuntimeShape& output_shape, float* output_data) { in LeakyRelu()
41 const RuntimeShape& output_shape, in QuantizeLeakyRelu()
/external/tensorflow/tensorflow/core/tpu/kernels/
Dimage_resize_ops.cc43 xla::Shape output_shape = in GetOutputShape() local
55 const xla::Shape& output_shape) { in CompileGrad()
77 auto output_shape = GetOutputShape(ctx); in CompileForward() local
131 auto output_shape = in Compile() local
/external/tensorflow/tensorflow/lite/kernels/
Dsparse_to_dense.cc40 TfLiteStatus Resize(TfLiteContext* context, const TfLiteTensor* output_shape, in Resize()
53 const TfLiteTensor* output_shape, in CheckDimensionsMatch()
129 const TfLiteTensor* output_shape, in ResizeOutputShape()
149 const TfLiteTensor* output_shape; in Prepare() local
205 const TfLiteTensor* output_shape; in SparseToDenseImpl() local
Daudio_spectrogram_test.cc72 std::vector<int> output_shape = m.GetOutputShape(); in TEST() local
88 std::vector<int> output_shape = m.GetOutputShape(); in TEST() local
104 std::vector<int> output_shape = m.GetOutputShape(); in TEST() local
Dreshape.cc37 TfLiteIntArray* output_shape = GetOutputShape(context, node); in ResizeOutput() local
78 TfLiteIntArray* output_shape = TfLiteIntArrayCreate(shape->dims->data[0]); in GetOutputShapeFromTensor() local
99 TfLiteIntArray* output_shape = TfLiteIntArrayCreate(num_dimensions); in GetOutputShapeFromParam() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dconvert_tensor_test.cc46 PartialTensorShape output_shape = in TEST() local
56 PartialTensorShape output_shape = ConvertTypeToTensorShape( in TEST() local
66 PartialTensorShape output_shape = ConvertTypeToTensorShape( in TEST() local
75 PartialTensorShape output_shape = ConvertTypeToTensorShape(b.getF32Type()); in TEST() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_layout_assignment.cc129 Shape* output_shape; in AddBackendConstraintsToDnnConvCustomCall() local
232 Shape output_shape = instruction->shape(); in AddBackendConstraints() local
244 Shape output_shape = instruction->shape(); in AddBackendConstraints() local
292 Shape output_shape = instruction->shape(); in AddBackendConstraints() local
/external/tensorflow/tensorflow/lite/delegates/coreml/builders/
Dreshape_op_builder.cc50 void ReshapeOpBuilder::SetShapeFromTensor(const TfLiteTensor* output_shape, in SetShapeFromTensor()
60 void ReshapeOpBuilder::SetShapeFromIntArray(const TfLiteIntArray* output_shape, in SetShapeFromIntArray()
100 TfLiteIntArray* output_shape = TfLiteIntArrayCreate(params->num_dimensions); in RegisterInputs() local
/external/tensorflow/tensorflow/python/keras/layers/
Deinsum_dense.py114 output_shape, argument
209 def _analyze_einsum_string(equation, bias_axes, input_shape, output_shape): argument
243 output_shape, argument
/external/tensorflow/tensorflow/core/kernels/
Dsegment_reduction_ops_impl.h113 TensorShape output_shape = input.shape(); in Compute() local
248 TensorShape output_shape = input.shape(); in ComputeAsync() local
288 TensorShape output_shape = input.shape(); in ComputeAsync() local

12345678910>>...18