Searched refs:argument_idx (Results 1 – 2 of 2) sorted by relevance
92 int64 argument_idx, in MustArgumentBeConstant() argument100 *result = absl::c_binary_search(constant_input_indices, argument_idx); in MustArgumentBeConstant()105 int64 argument_idx, in ShouldArgumentBeConstant() argument110 return MustArgumentBeConstant(op_kernel, argument_idx, result); in ShouldArgumentBeConstant()
41 Status ShouldArgumentBeConstant(const OpKernel* op_kernel, int64 argument_idx,43 Status MustArgumentBeConstant(const OpKernel* op_kernel, int64 argument_idx,