OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:x_input
(Results
1 - 3
of
3
) sorted by null
/external/tensorflow/tensorflow/core/kernels/
fused_batch_norm_op.h
65
const Tensor&
x_input
, const Tensor& scale_input,
74
typename TTypes<T, 4>::ConstTensor input(
x_input
.tensor<T, 4>());
cwise_ops_common.h
168
const Tensor&
x_input
= context->input(0);
variable
171
context,
x_input
.shape() == y_input.shape(),
173
"x shape: ",
x_input
.shape().DebugString(),
177
context->allocate_output(0,
x_input
.shape(), &z_output));
179
typename TTypes<T>::ConstFlat x(
x_input
.flat<T>());
fused_batch_norm_op.cc
50
void operator()(OpKernelContext* context, const Tensor&
x_input
,
61
typename TTypes<T, 4>::ConstTensor x(
x_input
.tensor<T, 4>());
135
const Tensor&
x_input
, const Tensor& scale_input,
145
typename TTypes<T, 4>::ConstTensor x(
x_input
.tensor<T, 4>());
490
const Tensor&
x_input
, const Tensor& scale_input, \
[
all
...]
Completed in 83 milliseconds