| /external/tensorflow/tensorflow/compiler/mlir/hlo/tests/ |
| D | lhlo-legalize-select-and-scatter.mlir | 9 func @select_and_scatter(%arg: memref<112x112xf32>,
13 "lmhlo.select_and_scatter"(%arg, %src, %init, %result) ( {
34 // CHECK-LABEL: func @select_and_scatter(
110 // The LHLO IR of the select block of the lhlo.select_and_scatter is applied
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dfs_hlo_visitor_with_default.h | 189 Status HandleSelectAndScatter(HloInstructionPtr select_and_scatter) override { in HandleSelectAndScatter() argument
190 return DefaultAction(select_and_scatter); in HandleSelectAndScatter()
|
| D | hlo_evaluator_typed_visitor.h | 1840 Status HandleSelectAndScatter(HloInstruction* select_and_scatter) override {
1841 auto operand = select_and_scatter->operand(0);
1842 auto source = select_and_scatter->operand(1);
1843 const Window& window = select_and_scatter->window();
1846 parent_->GetEvaluatedLiteralFor(select_and_scatter->operand(2));
1850 Literal result(select_and_scatter->shape());
1863 HloComputation* select = select_and_scatter->select();
1864 HloComputation* scatter = select_and_scatter->scatter();
1941 parent_->evaluated_[select_and_scatter] = std::move(result);
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | mhlo_to_lhlo_with_xla.cc | 584 TF_ASSIGN_OR_RETURN(auto select_and_scatter, in EmitSelectAndScatterOp()
592 select_and_scatter.window_dimensionsAttr( in EmitSelectAndScatterOp()
596 select_and_scatter.window_stridesAttr( in EmitSelectAndScatterOp()
600 select_and_scatter.paddingAttr( in EmitSelectAndScatterOp()
607 *select_and_scatter_instr->select(), &select_and_scatter.select(), in EmitSelectAndScatterOp()
610 *select_and_scatter_instr->scatter(), &select_and_scatter.scatter(), in EmitSelectAndScatterOp()
612 return select_and_scatter; in EmitSelectAndScatterOp()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 664 Status IrEmitter::HandleSelectAndScatter(HloInstruction* select_and_scatter) { in HandleSelectAndScatter() argument
665 CHECK_EQ(select_and_scatter->operand_count(), 3); in HandleSelectAndScatter()
666 const auto operand = select_and_scatter->operand(0); in HandleSelectAndScatter()
667 const auto source = select_and_scatter->operand(1); in HandleSelectAndScatter()
668 const auto init_value = select_and_scatter->operand(2); in HandleSelectAndScatter()
669 const Window& window = select_and_scatter->window(); in HandleSelectAndScatter()
703 select_and_scatter, /*desc=*/IrName(select_and_scatter, "init"), in HandleSelectAndScatter()
710 llvm_ir::ForLoopNest source_loops(IrName(select_and_scatter), &b_); in HandleSelectAndScatter()
729 llvm_ir::ForLoopNest window_loops(IrName(select_and_scatter, "window"), &b_); in HandleSelectAndScatter()
791 *select_and_scatter->select(), in HandleSelectAndScatter()
[all …]
|
| D | ir_emitter.h | 150 Status HandleSelectAndScatter(HloInstruction* select_and_scatter) override;
|
| /external/tensorflow/tensorflow/compiler/tf2xla/python/ |
| D | xla.py | 430 select_and_scatter = gen_xla_ops.xla_select_and_scatter variable
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | xla_ops_test.py | 475 return xla.select_and_scatter(
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 2179 HloInstruction* select_and_scatter) { in HandleSelectAndScatter() argument
2180 const Window& window = select_and_scatter->window(); in HandleSelectAndScatter()
2181 const auto* operand = select_and_scatter->operand(0); in HandleSelectAndScatter()
2182 const auto* source = select_and_scatter->operand(1); in HandleSelectAndScatter()
2193 TF_RETURN_IF_ERROR(AssertNonDeterminismIsOkay(select_and_scatter->name())); in HandleSelectAndScatter()
2195 TF_ASSIGN_OR_RETURN(auto input, GetMlirEmitterInput(select_and_scatter)); in HandleSelectAndScatter()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/ |
| D | lhlo_ops.td | 489 def LHLO_SelectAndScatterOp: LHLO_Op<"select_and_scatter", []>,
|
| D | hlo_ops.td | 1122 def HLO_SelectAndScatterOp: HLO_Op<"select_and_scatter",
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/hlo_to_lhlo_with_xla/ |
| D | hlo_text_to_lhlo_no_opt.hlotxt | 70 // CHECK: "lmhlo.select_and_scatter"(%{{.*}}, %{{.*}}, %[[GLOBAL_MEMREF]], %{{.*}})
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/ |
| D | export.mlir | 802 %1 = "mhlo.select_and_scatter"(%arg0, %arg1, %0) ( {
|
| D | import.hlotxt | 820 // CHECK: [[RESULT:%.*]] = "mhlo.select_and_scatter"([[INPUT]], [[SOURCE]], [[INIT_VAL]]) ( {
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/ |
| D | legalize-tf.mlir | 1377 // CHECK: %[[RESULT:.*]] = "mhlo.select_and_scatter"(%[[INPUT]], %[[GRAD]], %[[ZERO]]) ( {
1400 // CHECK: %[[RESULT:.*]] = "mhlo.select_and_scatter"(%[[INPUT]], %[[GRAD]], %[[ZERO]]) ( {
|