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

  /external/tensorflow/tensorflow/core/kernels/
batchtospace_op.cc 132 TensorShape internal_output_shape; local
146 internal_output_shape.AddDim(input_batch_size / block_shape_product);
162 internal_output_shape.AddDim(cropped_size);
174 internal_output_shape.AddDim(depth);
192 internal_output_shape.dim_sizes()), \
spacetobatch_op.cc 128 TensorShape internal_output_shape; local
143 internal_output_shape.AddDim(input_batch_size * block_shape_product);
161 internal_output_shape.AddDim(output_size);
173 internal_output_shape.AddDim(depth);
194 internal_output_shape.dim_sizes())))); \

Completed in 76 milliseconds