Searched refs:InputExpression (Results 1 – 7 of 7) sorted by relevance
152 const XlaExpression& InputExpression(int index);153 const XlaExpression& InputExpression(absl::string_view name);
70 const XlaExpression& XlaOpKernelContext::InputExpression(int index) { in InputExpression() function in tensorflow::XlaOpKernelContext74 const XlaExpression& XlaOpKernelContext::InputExpression( in InputExpression() function in tensorflow::XlaOpKernelContext80 return InputExpression(index).AsXlaOp(builder()); in Input()84 return InputExpression(name).AsXlaOp(builder()); in Input()141 XlaExpression e = InputExpression(index); in ConstantInputReshaped()
54 ctx->xla_context()->SetRetval(index_, ctx->InputExpression(0)); in Compile()
35 XlaExpression::Kind kind = ctx->InputExpression(0).kind(); in Compile()
28 return ctx->InputExpression(index).kind() == XlaExpression::Kind::kTensorList; in IsTensorListInput()
131 auto e = ctx->InputExpression(0); in Compile()
131 const XlaExpression& expression = ctx->InputExpression(i); in ConvertLoopInvariantsToConst()