Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/
H A Dmlir_xla_op_kernel.cc50 TF_ASSIGN_OR_RETURN(arg.constant_value, ctx->ConstantInputTensor(i)); in ContextToXlaArgs()
H A Dxla_op_kernel.cc188 TF_ASSIGN_OR_RETURN(Tensor constant, ConstantInputTensor(index, mode)); in ConstantInputReshaped()
509 StatusOr<Tensor> XlaOpKernelContext::ConstantInputTensor( in ConstantInputTensor() function in tensorflow::XlaOpKernelContext
H A Dxla_op_kernel.h212 StatusOr<Tensor> ConstantInputTensor(
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dlight_outside_compilation.cc125 TF_ASSIGN_OR_RETURN(Tensor input_tensor, ctx->ConstantInputTensor(i)); in CompileToCustomCallCallingTfKernel()