HomeSort by relevance Sort by last modified time
    Searched refs:shape_array (Results 1 - 6 of 6) sorted by null

  /external/tensorflow/tensorflow/lite/toco/graph_transformations/
convert_trivial_pack_to_reshape.cc 64 Array& shape_array = model->GetOrCreateArray(shape_array_name); local
65 *(shape_array.mutable_shape()->mutable_dims()) = {
68 shape_array.data_type = ArrayDataType::kInt32;
69 auto& shape_buffer = shape_array.GetMutableBuffer<ArrayDataType::kInt32>();
convert_expanddims_to_reshape.cc 75 Array& shape_array = model->GetOrCreateArray(shape_array_name); local
76 *(shape_array.mutable_shape()->mutable_dims()) = {
79 shape_array.data_type = ArrayDataType::kInt32;
80 auto& shape_buffer = shape_array.GetMutableBuffer<ArrayDataType::kInt32>();
convert_trivial_transpose_to_reshape.cc 95 Array& shape_array = model->GetOrCreateArray(shape_array_name); local
96 *(shape_array.mutable_shape()->mutable_dims()) = {
99 shape_array.data_type = ArrayDataType::kInt32;
100 auto& shape_buffer = shape_array.GetMutableBuffer<ArrayDataType::kInt32>();
propagate_fixed_sizes.cc 410 auto& shape_array = model->GetArray(op->inputs[1]); local
411 if (!shape_array.has_shape()) {
415 if (!shape_array.buffer) {
419 CHECK(shape_array.data_type == ArrayDataType::kInt32)
428 shape_array.GetBuffer<ArrayDataType::kInt32>().data;
    [all...]
  /external/tensorflow/tensorflow/core/kernels/hexagon/
graph_transferer.cc 475 std::array<int64, SHAPE_ARRAY_SIZE> shape_array; local
481 shape_array = BuildShapeArray(shape_handle, context);
489 const_node_info.add_shape(shape_array[0]);
490 const_node_info.add_shape(shape_array[1]);
491 const_node_info.add_shape(shape_array[2]);
492 const_node_info.add_shape(shape_array[3]);
618 std::array<int64, SHAPE_ARRAY_SIZE> shape_array; local
624 shape_array = BuildShapeArray(shape_handle, context);
632 shape_array = ToTensorShapeArray(shapes.at(0));
636 if (shape_array[0] == 1 && shape_array[1] == 1 && shape_array[2] == 1)
    [all...]
  /external/tensorflow/tensorflow/lite/toco/
export_tensorflow.cc 1052 const auto& shape_array = model.GetArray(src_op.inputs[1]); local
    [all...]

Completed in 186 milliseconds