HomeSort by relevance Sort by last modified time
    Searched defs:input_tensor_shape (Results 1 - 5 of 5) sorted by null

  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
depthtospace_op.cc 44 const TensorShape input_tensor_shape = ctx->InputShape(0); variable
45 int input_rank = input_tensor_shape.dims();
51 input_tensor_shape.dim_sizes();
spacetodepth_op.cc 44 const TensorShape input_tensor_shape = ctx->InputShape(0); variable
45 int input_rank = input_tensor_shape.dims();
51 input_tensor_shape.dim_sizes();
  /external/tensorflow/tensorflow/contrib/periodic_resample/kernels/
periodic_resample_op.h 103 const auto& input_tensor_shape = input_tensor.shape(); local
117 context, desired_shape[i] >= input_tensor_shape.dim_size(i),
122 i, " input tensor has size ", input_tensor_shape.dim_size(i),
  /external/tensorflow/tensorflow/core/kernels/
mkl_pooling_ops_common.h 156 const TensorShape& input_tensor_shape) {
159 this->data_format_tf_, input_tensor_shape);
201 TensorShape input_tensor_shape = input_tensor.shape(); local
205 : memory::desc(TFShapeToMklDnnDimsInNCHW(input_tensor_shape,
210 input_tensor_shape);
333 const TensorShape& input_tensor_shape) {
337 input_tensor_shape);
342 : TFShapeToMklDnnDimsInNCHW(input_tensor_shape,
  /external/tensorflow/tensorflow/compiler/xla/service/
hlo_instruction_test.cc 1275 Shape input_tensor_shape = ShapeUtil::MakeShape(F32, {50, 49, 48, 47, 46}); local
    [all...]

Completed in 320 milliseconds