Searched refs:InputXlaShape (Results 1 – 8 of 8) sorted by relevance
40 auto input_shape_or = ctx->InputXlaShape(0); in Compile()100 auto input_shape_or = ctx->InputXlaShape(0); in Compile()
113 auto input_xla_shape = ctx->InputXlaShape(0); in Compile()174 VLOG(2) << "Reshape from " << ctx->InputXlaShape(0)->ToString() in Compile()
104 ctx->InputXlaShape("input").ValueOrDie(), "", false, {}, in Compile()382 auto input_shape = ctx->InputXlaShape(0).ValueOrDie(); in Compile()
105 ctx->InputXlaShape(0)->is_dynamic_dimension(i)); in Compile()
140 xla::Shape input_xla_shape = ctx->InputXlaShape(0).ValueOrDie(); in EmitDynamicSlice()414 xla::Shape grad_shape = ctx->InputXlaShape(4).ValueOrDie(); in CompileAsDynamicUpdateSlice()
106 xla::StatusOr<xla::Shape> InputXlaShape(int index);107 xla::StatusOr<xla::Shape> InputXlaShape(absl::string_view name);
77 xla::StatusOr<xla::Shape> XlaOpKernelContext::InputXlaShape(int index) { in InputXlaShape() function in tensorflow::XlaOpKernelContext81 xla::StatusOr<xla::Shape> XlaOpKernelContext::InputXlaShape( in InputXlaShape() function in tensorflow::XlaOpKernelContext
95 ctx->InputXlaShape(0)->is_dynamic_dimension(i)); in Compile()