HomeSort by relevance Sort by last modified time
    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