/external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/ |
create_im2col_arrays.cc | 46 const int kheight = weights_shape.dims(1); local 48 if (kwidth == 1 && kheight == 1 && conv_op->stride_width == 1 && 63 LogName(*conv_op), kwidth, kheight, conv_op->stride_width,
|
identify_l2_pool.cc | 88 l2pool_op->kheight = avpool_op->kheight;
|
propagate_fixed_sizes.cc | 33 int kheight, int stride_width, int stride_height, 43 output_height = (input_height + stride_height - kheight) / stride_height; 53 0, ((output_height - 1) * stride_height + kheight - input_height) / 2); 166 const int kheight = weights_shape.dims(1); local 168 ComputeConvSizes(input_shape, output_depth, kwidth, kheight, op->stride_width, 181 input_depth * kheight * kwidth}); 222 const int kheight = weights_shape.dims(1); local 224 ComputeConvSizes(input_shape, output_depth, kwidth, kheight, op->stride_width, 699 ComputeConvSizes(input_shape, output_depth, op->kwidth, op->kheight, 716 ComputeConvSizes(input_shape, output_depth, op->kwidth, op->kheight, [all...] |
/external/tensorflow/tensorflow/contrib/lite/toco/tflite/ |
operator_test.cc | 190 op.kheight = 1080; 197 EXPECT_EQ(op.kheight, output_toco_op->kheight); 222 op.kheight = 1080; 229 EXPECT_EQ(op.kheight, output_toco_op->kheight); 271 op.kheight = 1080; 280 EXPECT_EQ(op.kheight, output_toco_op->kheight);
|
operator.cc | 43 op.kheight, activation_function); 52 op->kheight = options.filter_height(); 356 op.kheight, activation_function); 365 op->kheight = options.filter_height(); 406 op.kheight, activation_function); 415 op->kheight = options.filter_height(); [all...] |
/external/tensorflow/tensorflow/contrib/lite/toco/ |
model.h | 639 int kheight = 0; member in struct:toco::AveragePoolOperator 670 int kheight = 0; member in struct:toco::MaxPoolOperator 685 int kheight = 0; member in struct:toco::L2PoolOperator [all...] |
tooling_util.cc | [all...] |
export_tensorflow.cc | [all...] |
import_tensorflow.cc | [all...] |
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
optimized_ops.h | 637 const Dims<4>& input_dims, int w, int h, int b, int kheight, int kwidth, 647 const int ih_ungated_end = (ih_ungated_start + kheight); 723 int stride_height, int pad_width, int pad_height, int kheight, [all...] |