Home
last modified time | relevance | path

Searched refs:num_output_elements (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/lite/micro/kernels/
Dreshape.cc53 int num_output_elements = 1; in ReshapeOutput() local
61 num_output_elements *= value; in ReshapeOutput()
65 output_shape->data[stretch_dim] = num_input_elements / num_output_elements; in ReshapeOutput()
66 num_output_elements *= output_shape->data[stretch_dim]; in ReshapeOutput()
70 TF_LITE_ENSURE_EQ(context, num_input_elements, num_output_elements); in ReshapeOutput()
Dreduce.cc45 int num_output_elements; member
91 op_data->num_output_elements = NumElements(output); in PrepareMax()
/external/tensorflow/tensorflow/lite/kernels/
Dreshape.cc53 int num_output_elements = 1; in ResizeOutput() local
61 num_output_elements *= value; in ResizeOutput()
65 output_shape->data[stretch_dim] = num_input_elements / num_output_elements; in ResizeOutput()
66 num_output_elements *= output_shape->data[stretch_dim]; in ResizeOutput()
69 TF_LITE_ENSURE_EQ(context, num_input_elements, num_output_elements); in ResizeOutput()
/external/XNNPACK/test/
Dbinary-elementwise-operator-tester.h229 const size_t num_output_elements = in TestQS8() local
248 std::vector<int8_t> output(num_output_elements); in TestQS8()
249 std::vector<float> output_ref(num_output_elements); in TestQS8()
367 const size_t num_output_elements = in TestF16() local
386 std::vector<uint16_t> output(num_output_elements); in TestF16()
387 std::vector<float> output_ref(num_output_elements); in TestF16()
518 const size_t num_output_elements = in TestF32() local
537 std::vector<float> output(num_output_elements); in TestF32()
538 std::vector<float> output_ref(num_output_elements); in TestF32()
563 const float output_min = num_output_elements == 1 ? in TestF32()
[all …]
Dconstant-pad-operator-tester.h88 inline size_t num_output_elements() const { in num_output_elements() function
141 std::vector<uint32_t> output(num_output_elements()); in TestX32()
142 std::vector<uint32_t> output_ref(num_output_elements()); in TestX32()
/external/tensorflow/tensorflow/core/util/
Dbcast.h44 int64 num_output_elements = 1; in ComputeBatchIndices() local
52 for (int64 k = 0; k < (dim - 1) * num_output_elements; ++k) { in ComputeBatchIndices()
53 (*out_indices)[num_output_elements + k] = (*out_indices)[k] + incr; in ComputeBatchIndices()
55 num_output_elements *= dim; in ComputeBatchIndices()
/external/tensorflow/tensorflow/core/kernels/hexagon/
Dgraph_transferer.cc479 const int64 num_output_elements = context->Value(num_elements_dim); in RegisterConstantNode() local
480 data_size = max_bytes_per_data * num_output_elements; in RegisterConstantNode()
929 const int64 num_output_elements = context->Value(num_elements_dim); in AppendNodeOutputParams() local
930 data_size = max_bytes_per_data * num_output_elements; in AppendNodeOutputParams()
933 CHECK_EQ(shapes.at(i).num_elements(), num_output_elements); in AppendNodeOutputParams()
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_ops.cc820 const int num_output_elements = output_type.getNumElements(); in Verify() local
822 if (num_output_elements % z_out != 0) { in Verify()
828 if (num_input_elements / z_in != num_output_elements / z_out) { in Verify()
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/
Dops.mlir2365 // expected-error @+1 {{num_input_elements / z_in != num_output_elements / z_out}}