Searched refs:InputXlaType (Results 1 – 5 of 5) sorted by relevance
33 {context->InputXlaType("input")}, in Compile()49 {context->InputXlaType("keys"), context->InputXlaType("values")}, in Compile()
144 auto zero = xla::Zero(ctx->builder(), ctx->InputXlaType("begin")); in EmitDynamicSlice()151 OP_REQUIRES(ctx, ctx->InputXlaType("begin") == xla::S32, in EmitDynamicSlice()156 xla::ConstantR0WithType(ctx->builder(), ctx->InputXlaType("begin"), in EmitDynamicSlice()448 auto zero = xla::Zero(ctx->builder(), ctx->InputXlaType("begin")); in CompileAsDynamicUpdateSlice()
432 xla::PrimitiveType boxes_xla_type = context->InputXlaType("boxes"); in Compile()433 xla::PrimitiveType scores_xla_type = context->InputXlaType("scores"); in Compile()
90 xla::PrimitiveType InputXlaType(absl::string_view name);
120 xla::PrimitiveType XlaOpKernelContext::InputXlaType(absl::string_view name) { in InputXlaType() function in tensorflow::XlaOpKernelContext