Home
last modified time | relevance | path

Searched defs:shape_array (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/
H A Dconvert_trivial_pack_to_reshape.cc64 Array& shape_array = model->GetOrCreateArray(shape_array_name); in Run() local
H A Dconvert_expanddims_to_reshape.cc82 Array& shape_array = model->GetOrCreateArray(shape_array_name); in Run() local
H A Dconvert_trivial_transpose_to_reshape.cc97 Array& shape_array = model->GetOrCreateArray(shape_array_name); in Run() local
H A Didentify_nearest_upsample.cc197 Array& shape_array = model->GetOrCreateArray(shape_array_name); in Run() local
H A Dpropagate_fixed_sizes.cc411 auto& shape_array = model->GetArray(op->inputs[1]); in ProcessTensorFlowReshapeOperator() local
2127 auto& shape_array = model->GetArray(op->inputs[2]); in ProcessScatterNdOperator() local
/aosp_15_r20/external/ComputeLibrary/src/common/utils/
H A DLegacySupport.cpp79 int32_t *shape_array = new int32_t[num_dims]; in create_tensor_shape_array() local
/aosp_15_r20/external/pytorch/torch/csrc/profiler/standalone/
H A Dexecution_trace_observer.cpp406 std::vector<std::string> shape_array; in convertIValue() local
427 std::vector<std::string> shape_array; in convertIValue() local
/aosp_15_r20/external/tflite-support/tensorflow_lite_support/java/src/native/task/vision/segmenter/
H A Dimage_segmenter_jni.cc103 int shape_array[2] = {segmentation.height(), segmentation.width()}; in ConvertToSegmentationResults() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/
H A Dexport_tensorflow.cc1065 const auto& shape_array = model.GetArray(src_op.inputs[1]); in ConvertTensorFlowReshapeOperator() local