Searched refs:num_retries_per_sample_tensor (Results 1 – 1 of 1) sorted by relevance
87 const Tensor& num_retries_per_sample_tensor = context->input(3); in Compute() local98 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()