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()
|