Searched refs:InputIndex (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_op_kernel.cc | 119 static xla::StatusOr<int> InputIndex(XlaOpKernelContext* context, in InputIndex() function
134 TF_ASSIGN_OR_RETURN(int index, InputIndex(this, name)); in ConstantInput()
218 TF_ASSIGN_OR_RETURN(int index, InputIndex(this, name)); in ConstantInputAsIntScalar()
259 TF_ASSIGN_OR_RETURN(int index, InputIndex(this, name)); in ConstantInputAsIntVector()
273 TF_ASSIGN_OR_RETURN(int index, InputIndex(this, name)); in ConstantInputReshapedToIntVector()
307 TF_ASSIGN_OR_RETURN(int index, InputIndex(this, name)); in ConstantInputAsInt64Literal()
|