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

  /external/tensorflow/tensorflow/core/kernels/
batchtospace_op.cc 44 const Tensor& orig_input_tensor,
47 const int input_dims = orig_input_tensor.dims();
55 context, orig_input_tensor.dims() >= 1 + block_dims,
57 " instead of ", orig_input_tensor.dims()));
106 const int64 orig_input_batch_size = orig_input_tensor.dim_size(0);
122 context->set_output(0, orig_input_tensor);
141 const int64 size = orig_input_tensor.dim_size(block_dim + 1);
154 const int64 input_size = orig_input_tensor.dim_size(block_dim + 1);
169 const int64 size = orig_input_tensor.dim_size(dim);
194 orig_input_tensor.shaped<T, NUM_BLOCK_DIMS + 2>(
210 const Tensor& orig_input_tensor = context->input(0); variable
    [all...]
spacetobatch_op.cc 46 const Tensor& orig_input_tensor,
49 const int input_dims = orig_input_tensor.dims();
57 context, orig_input_tensor.dims() >= 1 + block_dims,
59 " instead of ", orig_input_tensor.dims()));
118 context->set_output(0, orig_input_tensor);
133 external_output_shape.AddDim(orig_input_tensor.dim_size(0) *
136 int64 input_batch_size = orig_input_tensor.dim_size(0);
138 const int64 size = orig_input_tensor.dim_size(block_dim + 1);
151 const int64 input_size = orig_input_tensor.dim_size(block_dim + 1);
168 const int64 size = orig_input_tensor.dim_size(dim)
211 const Tensor& orig_input_tensor = context->input(0); variable
    [all...]
mkl_maxpooling_op.cc 622 const Tensor& orig_input_tensor = variable
637 SanityCheckInputs(context, orig_input_tensor, orig_output_tensor,
651 context, orig_input_tensor, orig_input_mkl_shape,
742 const Tensor& orig_input_tensor,
751 OP_REQUIRES(context, orig_input_tensor.dims() == 4,
    [all...]

Completed in 223 milliseconds