Searched defs:shape_or (Results 1 – 8 of 8) sorted by relevance
113 auto shape_or = ctx->builder()->GetShape(ctx->Input(i + 1)); in Compile() local315 auto shape_or = b->GetShape(output_handle); in Compile() local338 auto shape_or = b->GetShape(token_output); in Compile() local
41 auto shape_or = ctx->builder()->GetShape(input); in Compile() local
203 auto shape_or = ctx->builder()->GetShape(ctx->Input(i + 1)); in Compile() local323 auto shape_or = b->GetShape(token_output); in Compile() local
484 auto shape_or = b->GetShape(buffer); in Compile() local530 auto shape_or = b->GetShape(input_tensor); in Compile() local
596 auto shape_or = builder->GetShape(token_output); in Compile() local
87 auto shape_or = b->GetShape(lhs_params[0]); in CreateScalarComparisonComputation() local
486 auto shape_or = expr.GetShape(); in LegalizeOp() local
1014 auto shape_or = sorted.builder()->GetShape(sorted); in ExportXlaOp() local