Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dconvert_trivial_pack_to_reshape.cc64 Array& shape_array = model->GetOrCreateArray(shape_array_name); in Run() local
Dconvert_expanddims_to_reshape.cc82 Array& shape_array = model->GetOrCreateArray(shape_array_name); in Run() local
Dconvert_trivial_transpose_to_reshape.cc96 Array& shape_array = model->GetOrCreateArray(shape_array_name); in Run() local
Didentify_nearest_upsample.cc196 Array& shape_array = model->GetOrCreateArray(shape_array_name); in Run() local
Dpropagate_fixed_sizes.cc411 auto& shape_array = model->GetArray(op->inputs[1]); in ProcessTensorFlowReshapeOperator() local
2126 auto& shape_array = model->GetArray(op->inputs[2]); in ProcessScatterNdOperator() local
/external/tflite-support/tensorflow_lite_support/java/src/native/task/vision/segmenter/
Dimage_segmenter_jni.cc103 int shape_array[2] = {segmentation.height(), segmentation.width()}; in ConvertToSegmentationResults() local
/external/tensorflow/tensorflow/core/kernels/hexagon/
Dgraph_transferer.cc475 std::array<int64, SHAPE_ARRAY_SIZE> shape_array; in RegisterConstantNode() local
618 std::array<int64, SHAPE_ARRAY_SIZE> shape_array; in IsNodeFlattenReshape() local
/external/tensorflow/tensorflow/lite/toco/
Dexport_tensorflow.cc1060 const auto& shape_array = model.GetArray(src_op.inputs[1]); in ConvertTensorFlowReshapeOperator() local