Searched refs:dims_size (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | tensor_ctypes.h | 40 const int dims_size = dims->size; in GetTensorShape() local
42 return RuntimeShape(dims_size, dims_data); in GetTensorShape()
|
| /external/tensorflow/tensorflow/lite/micro/examples/network_tester/ |
| D | network_tester_test.cc | 43 int dims_size = output->dims->size; in print_output_data() local
45 printf("\"dims\": [%d,", dims_size); in print_output_data()
49 printf("%d],\n", output->dims->data[dims_size - 1]); in print_output_data()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | kernel_util.cc | 35 const int dims_size = dims->size; in GetTensorShape() local
37 return RuntimeShape(dims_size, dims_data); in GetTensorShape()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | tooling_util.cc | 1496 num_dims = input_array.shape().dims_size(); in CreateOrCheckRnnStateArray()
1567 QCHECK_EQ(specified_input_array.shape().dims_size(), in ResolveModelFlags()
1568 dst_input_array->shape().dims_size()) in ResolveModelFlags()
1571 << specified_input_array.shape().dims_size() in ResolveModelFlags()
1574 << dst_input_array->shape().dims_size(); in ResolveModelFlags()
1578 for (int i = 1; i < specified_input_array.shape().dims_size(); i++) { in ResolveModelFlags()
1673 input_array_proto.shape().dims_size()); in ResolveModelFlags()
|
| /external/tensorflow/tensorflow/lite/core/ |
| D | subgraph.h | 468 TfLiteStatus BytesRequired(TfLiteType type, const int* dims, size_t dims_size,
|
| D | subgraph.cc | 649 size_t dims_size, size_t* bytes) { in BytesRequired() argument
652 for (int k = 0; k < dims_size; k++) { in BytesRequired()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | generic_layout_optimizer_transposer.cc | 620 const int dims_size = dims.size(); in IsFaninPortDimsNIfConst() local
621 if (tensor.dims() != dims_size) { in IsFaninPortDimsNIfConst()
624 for (int i = 0; i < dims_size; ++i) { in IsFaninPortDimsNIfConst()
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | propagate_fixed_sizes.cc | 2100 int dims_size = dims->size(); in ProcessMatrixDiagOperator() local
2102 CHECK_GT(dims_size, 0); in ProcessMatrixDiagOperator()
2103 int last_dim = (*dims)[dims_size - 1]; in ProcessMatrixDiagOperator()
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate.cc | 4601 const int dims_size = dims_tensor.dims->data[0]; in AddOpsAndTensors() local
4602 std::vector<int32_t> dims_int32(dims_size); in AddOpsAndTensors()
4603 std::copy(dims_tensor.data.i64, dims_tensor.data.i64 + dims_size, in AddOpsAndTensors()
|