Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dclustering_ops.cc87 const Tensor& num_retries_per_sample_tensor = context->input(3); in Compute() local
98 TensorShapeUtils::IsScalar(num_retries_per_sample_tensor.shape()), in Compute()
106 const int64_t value = num_retries_per_sample_tensor.scalar<int64_t>()(); in Compute()