Home
last modified time | relevance | path

Searched refs:x_input (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dlistdiff_op.cc73 std::vector<int64> x_input, y_input; in ListDiff() local
74 TF_RETURN_IF_ERROR(context->ConstantInputAsIntVector(0, &x_input)); in ListDiff()
85 auto x_size = x_input.size(); in ListDiff()
87 if (y_input_set.count(x_input[i]) > 0) { in ListDiff()
90 val_output.push_back(x_input[i]); in ListDiff()
/external/tensorflow/tensorflow/python/keras/engine/
Dtraining.py2385 x_input, y_input, _ = training_utils.extract_tensors_from_dataset(x)
2387 x_input = x
2391 if isinstance(x_input, (list, tuple)):
2393 tensor_util.is_tensor(v) for v in x_input):
2396 all_inputs += list(x_input)
2397 elif isinstance(x_input, dict):
2399 keys = sorted(x_input.keys())
2400 all_inputs = [x_input[k] for k in keys]
2402 if (not isinstance(x_input, np.ndarray) and
2403 not tensor_util.is_tensor(x_input)):
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dfused_batch_norm_op.h65 const Tensor& x_input, const Tensor& scale_input, in operator()
74 typename TTypes<T, 4>::ConstTensor input(x_input.tensor<T, 4>()); in operator()
Dcwise_ops_common.h170 const Tensor& x_input = context->input(0); in Compute() local
173 context, x_input.shape() == y_input.shape(), in Compute()
175 "x shape: ", x_input.shape().DebugString(), in Compute()
179 context->allocate_output(0, x_input.shape(), &z_output)); in Compute()
181 typename TTypes<T>::ConstFlat x(x_input.flat<T>()); in Compute()
Dfused_batch_norm_op.cc67 void operator()(OpKernelContext* context, const Tensor& x_input, in operator ()()
78 typename TTypes<T, 4>::ConstTensor x(x_input.tensor<T, 4>()); in operator ()()
156 const Tensor& x_input, const Tensor& scale_input, in operator ()()
166 typename TTypes<T, 4>::ConstTensor x(x_input.tensor<T, 4>()); in operator ()()
511 const Tensor& x_input, const Tensor& scale_input, \