Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dreshape.cc51 int64_t non_zero_num_input_elements = 1, num_input_elements = 1; in ResizeOutput() local
55 num_input_elements *= value; in ResizeOutput()
76 if (num_input_elements == 0 && num_output_elements != 0) { in ResizeOutput()
85 TF_LITE_ENSURE_EQ(context, num_input_elements, num_output_elements); in ResizeOutput()
/aosp_15_r20/external/XNNPACK/src/subgraph/
H A Dstatic-reshape.c185 const size_t num_input_elements = xnn_shape_multiply_all_dims(&input_value->shape); in xnn_define_static_reshape() local
188 if (num_input_elements != num_output_elements) { in xnn_define_static_reshape()
192 … xnn_node_type_to_string(xnn_node_type_static_reshape), input_id, output_id, num_input_elements, in xnn_define_static_reshape()
H A Dfully-connected.c41 const size_t num_input_elements = xnn_shape_multiply_all_dims(&values[node->inputs[0]].shape); in create_fully_connected_operator() local
151 opdata->batch_size = num_input_elements / input_channels; in create_fully_connected_operator()
/aosp_15_r20/external/tensorflow/tensorflow/core/util/
H A Dbcast.h45 int64_t num_input_elements = 1; in ComputeBatchIndices() local
51 const int64_t incr = bcast[i] > 1 ? 0 : num_input_elements; in ComputeBatchIndices()
56 num_input_elements *= reshape[i]; in ComputeBatchIndices()
/aosp_15_r20/external/XNNPACK/test/
H A Dconstant-pad-operator-tester.h42 inline size_t num_input_elements() const { in num_input_elements() function
140 std::vector<uint8_t> input(XNN_EXTRA_BYTES / sizeof(uint8_t) + num_input_elements()); in TestX8()
255 std::vector<uint16_t> input(XNN_EXTRA_BYTES / sizeof(uint16_t) + num_input_elements()); in TestX16()
370 std::vector<uint32_t> input(XNN_EXTRA_BYTES / sizeof(uint32_t) + num_input_elements()); in TestX32()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dshape_inference.cc1171 int num_input_elements; in InferShapeForReduceDataset() local
1175 num_input_elements = input_elements.shapes.size(); in InferShapeForReduceDataset()
1177 num_input_elements = in InferShapeForReduceDataset()
1183 << num_states << " , #input_elements = " << num_input_elements in InferShapeForReduceDataset()
1185 if (num_states + num_input_elements + num_captured_arguments != in InferShapeForReduceDataset()
1208 for (int i = 0; i < num_input_elements; ++i) { in InferShapeForReduceDataset()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/xnnpack/
H A Dxnnpack_delegate.cc3019 int32_t num_input_elements = 1; in VisitFullyConnectedNode() local
3027 num_input_elements *= SizeOfDimension(&input_tensor, i); in VisitFullyConnectedNode()
3048 if (num_input_elements % input_channels != 0) { in VisitFullyConnectedNode()
3061 num_input_elements / input_channels) { in VisitFullyConnectedNode()
3067 num_input_elements / input_channels, node->inputs->data[0]); in VisitFullyConnectedNode()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
H A Dtfl_ops.cc1033 const int num_input_elements = input_type.getNumElements(); in verify() local
1034 if (z_in != 0 && num_input_elements % z_in != 0) { in verify()
1057 if (z_in != 0 && num_input_elements / z_in != num_output_elements / z_out) { in verify()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/tests/
H A Dops.mlir2484 // expected-error @+1 {{num_input_elements / z_in != num_output_elements / z_out}}