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

  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
image_ops.cc 167 int height_dim = input_shape.dims() - 3; variable
170 const int64 height = input_shape.dim_size(height_dim);
185 {height_dim, width_dim});
  /external/tensorflow/tensorflow/core/kernels/
bias_op.cc 59 int32 height_dim = value_tensor.dims() - 2; local
62 *height = static_cast<int32>(value_tensor.dim_size(height_dim));

Completed in 108 milliseconds