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

  /external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/
resolve_reshape_attributes.cc 40 const auto& constant_input_array = model->GetArray(reshape_op->inputs[1]); local
41 op->shape = constant_input_array.GetBuffer<ArrayDataType::kInt32>().data;
resolve_multiply_by_zero.cc 89 const auto& constant_input_array = local
92 CHECK(constant_input_array.data_type == output_array.data_type);
96 constant_input_array.GetBuffer<ArrayDataType::kFloat>().data;
105 constant_input_array.GetBuffer<ArrayDataType::kUint8>().data;
114 constant_input_array.GetBuffer<ArrayDataType::kInt32>().data;
123 constant_input_array.GetBuffer<ArrayDataType::kInt64>().data;
remove_trivial_binary.cc 83 const auto& constant_input_array = local
86 if (constant_input_array.data_type != ArrayDataType::kFloat) {
90 constant_input_array.GetBuffer<ArrayDataType::kFloat>().data;

Completed in 181 milliseconds