Searched defs:block_shape_array (Results 1 - 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/
H A Dresolve_batch_to_space_nd_attributes.cc61 const auto& block_shape_array = model->GetArray(op->inputs[1]); local
62 if (!block_shape_array.has_shape()) return false;
63 const std::vector<int>& block_shape_dims = block_shape_array.shape().dims();
66 block_shape_array.GetBuffer<ArrayDataType::kInt32>().data;
H A Dresolve_space_to_batch_nd_attributes.cc64 const auto& block_shape_array = local
66 if (!block_shape_array.has_shape()) return false;
67 const std::vector<int>& block_shape_dims = block_shape_array.shape().dims();
70 block_shape_array.GetBuffer<ArrayDataType::kInt32>().data;
H A Dpropagate_fixed_sizes.cc862 const auto& block_shape_array = model->GetArray(op->inputs[1]); local
864 const auto& block_shape_array_shape = block_shape_array.shape();
871 if (!block_shape_array.buffer) {
874 QCHECK(block_shape_array.data_type == ArrayDataType::kInt32);
876 block_shape_array.GetBuffer<ArrayDataType::kInt32>().data;
911 const auto& block_shape_array = model->GetArray(op->inputs[1]); local
913 const auto& block_shape_array_shape = block_shape_array.shape();
920 if (!block_shape_array.buffer) {
923 QCHECK(block_shape_array.data_type == ArrayDataType::kInt32);
925 block_shape_array
[all...]

Completed in 100 milliseconds