Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_op_kernel.cc119 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()