Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_op_kernel.h152 const XlaExpression& InputExpression(int index);
153 const XlaExpression& InputExpression(absl::string_view name);
Dxla_op_kernel.cc70 const XlaExpression& XlaOpKernelContext::InputExpression(int index) { in InputExpression() function in tensorflow::XlaOpKernelContext
74 const XlaExpression& XlaOpKernelContext::InputExpression( in InputExpression() function in tensorflow::XlaOpKernelContext
80 return InputExpression(index).AsXlaOp(builder()); in Input()
84 return InputExpression(name).AsXlaOp(builder()); in Input()
141 XlaExpression e = InputExpression(index); in ConstantInputReshaped()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dretval_op.cc54 ctx->xla_context()->SetRetval(index_, ctx->InputExpression(0)); in Compile()
Daggregate_ops.cc35 XlaExpression::Kind kind = ctx->InputExpression(0).kind(); in Compile()
Dtensor_list_utils.cc28 return ctx->InputExpression(index).kind() == XlaExpression::Kind::kTensorList; in IsTensorListInput()
Dtranspose_op.cc131 auto e = ctx->InputExpression(0); in Compile()
Dwhile_op.cc131 const XlaExpression& expression = ctx->InputExpression(i); in ConvertLoopInvariantsToConst()