Searched refs:shape_or (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | sharding_op.cc | 41 auto shape_or = ctx->builder()->GetShape(input); in Compile() local
42 OP_REQUIRES_OK(ctx, shape_or.status()); in Compile()
46 {input}, shape_or.ValueOrDie())); in Compile()
|
| D | case_op.cc | 113 auto shape_or = ctx->builder()->GetShape(ctx->Input(i + 1)); in Compile() local
114 OP_REQUIRES_OK(ctx, shape_or.status()); in Compile()
115 arg.shape = shape_or.ValueOrDie(); in Compile()
315 auto shape_or = b->GetShape(output_handle); in Compile() local
316 if (shape_or.ok()) { in Compile()
318 << xla::ShapeUtil::HumanString(shape_or.ValueOrDie()); in Compile()
338 auto shape_or = b->GetShape(token_output); in Compile() local
339 OP_REQUIRES_OK(ctx, shape_or.status()); in Compile()
340 OP_REQUIRES(ctx, shape_or.ValueOrDie().IsToken(), in Compile()
343 xla::ShapeUtil::HumanString(shape_or.ValueOrDie()))); in Compile()
|
| D | if_op.cc | 203 auto shape_or = ctx->builder()->GetShape(ctx->Input(i + 1)); in Compile() local
204 OP_REQUIRES_OK(ctx, shape_or.status()); in Compile()
205 arg.shape = shape_or.ValueOrDie(); in Compile()
323 auto shape_or = b->GetShape(token_output); in Compile() local
324 OP_REQUIRES_OK(ctx, shape_or.status()); in Compile()
325 OP_REQUIRES(ctx, shape_or.ValueOrDie().IsToken(), in Compile()
328 xla::ShapeUtil::HumanString(shape_or.ValueOrDie()))); in Compile()
|
| D | tensor_list_ops.cc | 484 auto shape_or = b->GetShape(buffer); in Compile() local
485 OP_REQUIRES_OK(ctx, shape_or.status()); in Compile()
486 xla::Shape element_shape = shape_or.ConsumeValueOrDie(); in Compile()
530 auto shape_or = b->GetShape(input_tensor); in Compile() local
531 OP_REQUIRES_OK(ctx, shape_or.status()); in Compile()
532 xla::Shape element_shape = shape_or.ConsumeValueOrDie(); in Compile()
|
| D | while_op.cc | 596 auto shape_or = builder->GetShape(token_output); in Compile() local
597 OP_REQUIRES_OK(ctx, shape_or.status()); in Compile()
598 OP_REQUIRES(ctx, shape_or.ValueOrDie().IsToken(), in Compile()
601 xla::ShapeUtil::HumanString(shape_or.ValueOrDie()))); in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | comparators.cc | 87 auto shape_or = b->GetShape(lhs_params[0]); in CreateScalarComparisonComputation() local
88 if (!shape_or.ok()) { in CreateScalarComparisonComputation()
89 b->ReportError(shape_or.status()); in CreateScalarComparisonComputation()
92 Shape shape = shape_or.ValueOrDie(); in CreateScalarComparisonComputation()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf_with_tf2xla.cc | 486 auto shape_or = expr.GetShape(); in LegalizeOp() local
487 if (!shape_or.ok()) { in LegalizeOp()
494 shape_or.ValueOrDie()); in LegalizeOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | mlir_hlo_to_hlo.cc | 1014 auto shape_or = sorted.builder()->GetShape(sorted); in ExportXlaOp() local
1015 if (!shape_or.ok()) { in ExportXlaOp()
1016 return op.emitError(shape_or.status().ToString()); in ExportXlaOp()
1019 xla::Shape& shape = shape_or.ValueOrDie(); in ExportXlaOp()
|