Home
last modified time | relevance | path

Searched refs:shape_array (Results 1 – 6 of 6) 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
65 *(shape_array.mutable_shape()->mutable_dims()) = { in Run()
68 shape_array.data_type = ArrayDataType::kInt32; in Run()
69 auto& shape_buffer = shape_array.GetMutableBuffer<ArrayDataType::kInt32>(); in Run()
Dconvert_expanddims_to_reshape.cc75 Array& shape_array = model->GetOrCreateArray(shape_array_name); in Run() local
76 *(shape_array.mutable_shape()->mutable_dims()) = { in Run()
79 shape_array.data_type = ArrayDataType::kInt32; in Run()
80 auto& shape_buffer = shape_array.GetMutableBuffer<ArrayDataType::kInt32>(); in Run()
Dconvert_trivial_transpose_to_reshape.cc95 Array& shape_array = model->GetOrCreateArray(shape_array_name); in Run() local
96 *(shape_array.mutable_shape()->mutable_dims()) = { in Run()
99 shape_array.data_type = ArrayDataType::kInt32; in Run()
100 auto& shape_buffer = shape_array.GetMutableBuffer<ArrayDataType::kInt32>(); in Run()
Dpropagate_fixed_sizes.cc410 auto& shape_array = model->GetArray(op->inputs[1]); in ProcessTensorFlowReshapeOperator() local
411 if (!shape_array.has_shape()) { in ProcessTensorFlowReshapeOperator()
415 if (!shape_array.buffer) { in ProcessTensorFlowReshapeOperator()
419 CHECK(shape_array.data_type == ArrayDataType::kInt32) in ProcessTensorFlowReshapeOperator()
428 shape_array.GetBuffer<ArrayDataType::kInt32>().data; in ProcessTensorFlowReshapeOperator()
/external/tensorflow/tensorflow/core/kernels/hexagon/
Dgraph_transferer.cc475 std::array<int64, SHAPE_ARRAY_SIZE> shape_array; in RegisterConstantNode() local
481 shape_array = BuildShapeArray(shape_handle, context); in RegisterConstantNode()
489 const_node_info.add_shape(shape_array[0]); in RegisterConstantNode()
490 const_node_info.add_shape(shape_array[1]); in RegisterConstantNode()
491 const_node_info.add_shape(shape_array[2]); in RegisterConstantNode()
492 const_node_info.add_shape(shape_array[3]); in RegisterConstantNode()
618 std::array<int64, SHAPE_ARRAY_SIZE> shape_array; in IsNodeFlattenReshape() local
624 shape_array = BuildShapeArray(shape_handle, context); in IsNodeFlattenReshape()
632 shape_array = ToTensorShapeArray(shapes.at(0)); in IsNodeFlattenReshape()
636 if (shape_array[0] == 1 && shape_array[1] == 1 && shape_array[2] == 1) { in IsNodeFlattenReshape()
/external/tensorflow/tensorflow/lite/toco/
Dexport_tensorflow.cc1052 const auto& shape_array = model.GetArray(src_op.inputs[1]); in ConvertTensorFlowReshapeOperator() local
1053 QCHECK(shape_array.data_type == ArrayDataType::kInt32) in ConvertTensorFlowReshapeOperator()
1055 QCHECK(shape_array.buffer != nullptr) in ConvertTensorFlowReshapeOperator()
1057 const auto& shape_data = shape_array.GetBuffer<ArrayDataType::kInt32>().data; in ConvertTensorFlowReshapeOperator()