Searched defs:shape_or (Results 1 – 8 of 8) sorted by relevance
118 auto shape_or = ctx->builder()->GetShape(ctx->Input(i + 1)); in Compile() local306 auto shape_or = b->GetShape(output_handle); in Compile() local329 auto shape_or = b->GetShape(token_output); in Compile() local
222 auto shape_or = ctx->builder()->GetShape(ctx->Input(i + 1)); in Compile() local347 auto shape_or = b->GetShape(token_output); in Compile() local
492 auto shape_or = b->GetShape(buffer); in Compile() local538 auto shape_or = b->GetShape(input_tensor); in Compile() local
48 auto shape_or = ctx->builder()->GetShape(input); in Compile() local
609 auto shape_or = builder->GetShape(token_output); in Compile() local
87 auto shape_or = b->GetShape(lhs_params[0]); in CreateScalarComparisonComputation() local
626 auto shape_or = expr.GetShape(); in LegalizeOp() local
1580 auto shape_or = sorted.builder()->GetShape(sorted); in ExportXlaOp() local1644 auto shape_or = whileop.builder()->GetShape(whileop); in ExportXlaOp() local