Home
last modified time | relevance | path

Searched defs:dims_size (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/
H A Dtensor_ctypes.h40 const int dims_size = dims->size; in GetTensorShape() local
/aosp_15_r20/external/pytorch/binaries/
H A Daot_model_compiler.cc109 const auto dims_size = split(';', FLAGS_input_dims).size(); in main() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dreduce.cc171 TfLiteIntArray* dims_size = TfLiteIntArrayCreate(1); in ResizeTempDims() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/core/
H A Dsubgraph.cc689 size_t dims_size, size_t* bytes) { in BytesRequired()
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/
H A Dpropagate_fixed_sizes.cc2101 int dims_size = dims->size(); in ProcessMatrixDiagOperator() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dgeneric_layout_optimizer_transposer.cc615 const int dims_size = dims.size(); in IsFaninPortDimsNIfConst() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/nnapi/
H A Dnnapi_delegate.cc6000 const int dims_size = dims_tensor.dims->data[0]; in AddOpsAndTensors() local