| /external/tensorflow/tensorflow/core/kernels/ |
| spacetobatch_op.cc | 46 const Tensor& orig_input_tensor,
49 const int input_dims = orig_input_tensor.dims();
56 if (orig_input_tensor.dims() < 1 + block_dims) {
58 " instead of ", orig_input_tensor.dims());
119 context->set_output(0, orig_input_tensor);
134 external_output_shape.AddDim(orig_input_tensor.dim_size(0) *
137 int64 input_batch_size = orig_input_tensor.dim_size(0);
139 const int64 size = orig_input_tensor.dim_size(block_dim + 1);
153 const int64 input_size = orig_input_tensor.dim_size(block_dim + 1);
171 const int64 size = orig_input_tensor.dim_size(dim)
214 const Tensor& orig_input_tensor = context->input(0); variable
[all...] |
| 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...] |
| mkl_lrn_op.cc | 374 // Naming: diff_dst is input_gradient_tensor; src is orig_input_tensor.
376 const Tensor& orig_input_tensor = MklGetInput(context, kIdxOrigInput); variable
388 orig_input_dnn_data.SetUsrMem(orig_input_md, &orig_input_tensor);
524 Tensor orig_input_tensor; local
541 orig_input_tensor = ConvertMklToTF<T>(
544 orig_input_tensor = MklGetInput(context, kIdxOrigInput);
563 auto in_shaped = orig_input_tensor.shaped<T, 2>({nodes * batch, depth});
608 const Tensor& orig_input_tensor = MklGetInput(context, kIdxOrigInput); local
632 OP_REQUIRES(context, orig_input_tensor.dims() == 4,
|
| mkl_maxpooling_op.cc | 689 const Tensor& orig_input_tensor = variable
704 TensorShape orig_input_shape = orig_input_tensor.shape();
[all...] |
| mkl_avgpooling_op.cc | 576 const Tensor& orig_input_tensor = variable
589 auto shape_vec = orig_input_tensor.vec<int32>();
591 for (int i = 0; i < orig_input_tensor.NumElements(); i++) {
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| conv_ops_3d_test.py | 417 orig_input_tensor = constant_op.constant(
423 input_tensor = test_util.NHWCToNCHW(orig_input_tensor)
426 input_tensor = orig_input_tensor
444 orig_input_tensor, input_shape, conv, output_shape)
|
