Searched defs:input_shape_or (Results 1 – 3 of 3) sorted by relevance
40 auto input_shape_or = ctx->InputXlaShape(0); in Compile() local100 auto input_shape_or = ctx->InputXlaShape(0); in Compile() local
217 xla::StatusOr<xla::Shape> input_shape_or = ctx->builder()->GetShape(input); in Compile() local
504 auto input_shape_or = ctx->builder()->GetShape(input); in Compile() local