Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/jit/
Dxla_compile_on_demand_op.cc92 int64 argument_idx, in MustArgumentBeConstant() argument
100 *result = absl::c_binary_search(constant_input_indices, argument_idx); in MustArgumentBeConstant()
105 int64 argument_idx, in ShouldArgumentBeConstant() argument
110 return MustArgumentBeConstant(op_kernel, argument_idx, result); in ShouldArgumentBeConstant()
Dxla_compile_on_demand_op.h41 Status ShouldArgumentBeConstant(const OpKernel* op_kernel, int64 argument_idx,
43 Status MustArgumentBeConstant(const OpKernel* op_kernel, int64 argument_idx,