OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:orig_input_tensor
(Results
1 - 5
of
5
) sorted by null
/external/tensorflow/tensorflow/core/kernels/
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
...]
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_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
...]
mkl_lrn_op.cc
[
all
...]
/external/tensorflow/tensorflow/python/kernel_tests/
conv_ops_3d_test.py
324
orig_input_tensor
= constant_op.constant(
330
input_tensor = test_util.NHWCToNCHW(
orig_input_tensor
)
333
input_tensor =
orig_input_tensor
349
orig_input_tensor
, input_shape, conv, output_shape)
Completed in 111 milliseconds