| /external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/ |
| resolve_constant_range.cc | 61 CHECK_EQ(RequiredBufferSizeForShape(start_array.shape()), 1)
63 CHECK_EQ(RequiredBufferSizeForShape(limit_array.shape()), 1)
65 CHECK_EQ(RequiredBufferSizeForShape(delta_array.shape()), 1)
|
| read_fake_quant_min_max.cc | 68 CHECK_EQ(RequiredBufferSizeForShape(min_array.shape()), 1);
69 CHECK_EQ(RequiredBufferSizeForShape(max_array.shape()), 1);
|
| resolve_constant_fill.cc | 35 data.resize(RequiredBufferSizeForShape(output_array.shape()));
75 CHECK_EQ(RequiredBufferSizeForShape(val_array.shape()), 1);
|
| resolve_constant_stack.cc | 34 output_data.resize(RequiredBufferSizeForShape(output_array.shape()));
42 int input_size = RequiredBufferSizeForShape(input_array.shape());
|
| resolve_strided_slice_attributes.cc | 36 if (toco::RequiredBufferSizeForShape(start_array.shape()) > 4) {
|
| fuse_binary_into_following_affine.cc | 52 CHECK_EQ(RequiredBufferSizeForShape(operand.shape()), 1);
82 const int weights_size = RequiredBufferSizeForShape(weights_shape);
97 const int weights_size = RequiredBufferSizeForShape(weights_shape);
132 CHECK_EQ(RequiredBufferSizeForShape(operand.shape()), 1);
139 const int weights_size = RequiredBufferSizeForShape(weights.shape());
|
| convert_expanddims_to_reshape.cc | 54 CHECK_EQ(RequiredBufferSizeForShape(axis_array.shape()), 1);
|
| resolve_reorder_axes.cc | 39 reordered_data.resize(RequiredBufferSizeForShape(output_array->shape()));
|
| identify_relu1.cc | 43 RequiredBufferSizeForShape(op_array.shape()) != 1) {
|
| lstm_utils.cc | 50 buffer->data.resize(RequiredBufferSizeForShape(shape));
|
| resolve_multiply_by_zero.cc | 43 data.resize(RequiredBufferSizeForShape(array->shape()));
|
| resolve_constant_unary.cc | 109 const int output_buffer_size = RequiredBufferSizeForShape(output_shape);
115 const int input_buffer_size = RequiredBufferSizeForShape(input_shape);
|
| identify_l2_normalization.cc | 92 if (RequiredBufferSizeForShape(input_array.shape()) != 1) {
|
| resolve_batch_normalization.cc | 89 const int buffer_size = RequiredBufferSizeForShape(param_shape);
|
| resolve_constant_strided_slice.cc | 83 output_data.resize(RequiredBufferSizeForShape(output_array->shape()));
|
| propagate_fixed_sizes.cc | 296 CHECK_LE(RequiredBufferSizeForShape(dims_array.shape()), 4)
327 const int input_overall_size = RequiredBufferSizeForShape(input_shape);
382 const int input_flat_size = RequiredBufferSizeForShape(input_shape);
392 CHECK_EQ(input_flat_size, RequiredBufferSizeForShape(output_shape))
626 CHECK_EQ(RequiredBufferSizeForShape(start_array.shape()), 1)
628 CHECK_EQ(RequiredBufferSizeForShape(limit_array.shape()), 1)
630 CHECK_EQ(RequiredBufferSizeForShape(delta_array.shape()), 1)
667 CHECK_EQ(RequiredBufferSizeForShape(axis_array.shape()), 1)
[all...] |
| resolve_constant_transpose.cc | 38 output_data.resize(RequiredBufferSizeForShape(output_shape));
|
| fuse_binary_into_preceding_affine.cc | 135 const int weights_size = RequiredBufferSizeForShape(weights_shape);
|
| resolve_constant_binary.cc | 92 const int output_buffer_size = RequiredBufferSizeForShape(output_shape);
|
| /external/tensorflow/tensorflow/contrib/lite/toco/ |
| tooling_util.cc | 619 int RequiredBufferSizeForShape(const Shape& shape) {
[all...] |
| tooling_util.h | 129 int RequiredBufferSizeForShape(const Shape& shape);
235 DCHECK_LT(index, RequiredBufferSizeForShape(shape));
|
| allocate_transient_arrays.cc | 170 elem_size * RequiredBufferSizeForShape(array->shape());
|
| dump_graphviz.cc | 169 int buffer_size = RequiredBufferSizeForShape(array.shape());
|
| import_tensorflow.cc | 177 output_float_data.resize(RequiredBufferSizeForShape(output_array->shape()),
208 output_int_data.resize(RequiredBufferSizeForShape(output_array->shape()), 0);
234 output_int_data.resize(RequiredBufferSizeForShape(output_array->shape()), 0);
260 output_int_data.resize(RequiredBufferSizeForShape(output_array->shape()), 0);
286 output_string_data.resize(RequiredBufferSizeForShape(output_array->shape()));
[all...] |
| export_tensorflow.cc | 93 const int input_flat_size = RequiredBufferSizeForShape(input_shape);
115 const int input_flat_size = RequiredBufferSizeForShape(input_shape);
[all...] |
