Searched refs:InputXlaType (Results 1 – 6 of 6) sorted by relevance
38 {context->InputXlaType("keys"), context->InputXlaType("values")}, in Compile()
151 auto zero = xla::Zero(ctx->builder(), ctx->InputXlaType("begin")); in EmitDynamicSlice()158 OP_REQUIRES(ctx, ctx->InputXlaType("begin") == xla::S32, in EmitDynamicSlice()163 xla::ConstantR0WithType(ctx->builder(), ctx->InputXlaType("begin"), in EmitDynamicSlice()485 auto zero = xla::Zero(ctx->builder(), ctx->InputXlaType("begin")); in CompileAsDynamicUpdateSlice()
59 xla::PrimitiveType dtype = ctx->InputXlaType("weights"); in Compile()
448 xla::PrimitiveType boxes_xla_type = context->InputXlaType("boxes"); in ComputeResult()449 xla::PrimitiveType scores_xla_type = context->InputXlaType("scores"); in ComputeResult()
93 xla::PrimitiveType InputXlaType(absl::string_view name);
129 xla::PrimitiveType XlaOpKernelContext::InputXlaType(absl::string_view name) { in InputXlaType() function in tensorflow::XlaOpKernelContext