| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | dot_operation_test.cc | 965 auto dynamic_slice = DynamicSlice(lhs_constant, {one, zero}, {1, 6}); in XLA_TEST_F() local
970 DotGeneral(dynamic_slice, rhs_constant, dot_dnums); in XLA_TEST_F()
993 auto dynamic_slice = DynamicSlice(rhs_constant, {zero, one}, {6, 1}); in XLA_TEST_F() local
998 DotGeneral(lhs_constant, dynamic_slice, dot_dnums); in XLA_TEST_F()
1023 auto dynamic_slice = DynamicSlice(lhs_constant, {zero, one}, {6, 1}); in XLA_TEST_F() local
1028 DotGeneral(dynamic_slice, rhs_constant, dot_dnums); in XLA_TEST_F()
1051 auto dynamic_slice = DynamicSlice(rhs_constant, {one, zero}, {1, 6}); in XLA_TEST_F() local
1056 DotGeneral(lhs_constant, dynamic_slice, dot_dnums); in XLA_TEST_F()
1084 auto dynamic_slice = DynamicSlice(lhs_constant, {zero, one}, {6, 1}); in XLA_TEST_F() local
1089 DotGeneral(dynamic_slice, rhs_constant, dot_dnums); in XLA_TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_cost_analysis.cc | 260 const HloInstruction* dynamic_slice) { in HandleDynamicSlice() argument
262 GetShapeSize(dynamic_slice->shape()) * 2 + in HandleDynamicSlice()
263 GetShapeSize(dynamic_slice->operand(1)->shape()); in HandleDynamicSlice()
264 SetOutputBytesAccessed(GetShapeSize(dynamic_slice->shape())); in HandleDynamicSlice()
265 SetOperandBytesAccessed(0, GetShapeSize(dynamic_slice->shape())); in HandleDynamicSlice()
266 SetOperandBytesAccessed(1, GetShapeSize(dynamic_slice->operand(1)->shape())); in HandleDynamicSlice()
|
| D | dfs_hlo_visitor_with_default.h | 170 Status HandleDynamicSlice(HloInstructionPtr dynamic_slice) override { in HandleDynamicSlice() argument
171 return DefaultAction(dynamic_slice); in HandleDynamicSlice()
|
| D | algebraic_simplifier.cc | 355 Status HandleDynamicSlice(HloInstruction* dynamic_slice) override;
2032 HloDynamicSliceInstruction* dynamic_slice = in OptimizeDotOfGather() local
2061 auto indices_type = dynamic_slice->operand(1)->shape().element_type(); in OptimizeDotOfGather()
2067 dynamic_slice->mutable_operand(1 + index_of_non_zero_start); in OptimizeDotOfGather()
2069 dynamic_slice->mutable_operand(1 + index_of_zero_start); in OptimizeDotOfGather()
4342 HloInstruction* dynamic_slice) { in HandleDynamicSlice() argument
4343 auto operand = dynamic_slice->mutable_operand(0); in HandleDynamicSlice()
4344 if (ShapeUtil::IsScalar(dynamic_slice->shape())) { in HandleDynamicSlice()
4345 return ReplaceInstruction(dynamic_slice, operand); in HandleDynamicSlice()
4349 if (SameShape(operand, dynamic_slice)) { in HandleDynamicSlice()
[all …]
|
| D | hlo_evaluator_typed_visitor.h | 1600 Status HandleDynamicSlice(HloInstruction* dynamic_slice) override {
1601 auto operand = dynamic_slice->operand(0);
1602 auto start_indices = dynamic_slice->operand(1);
1603 auto result_shape = dynamic_slice->shape();
1608 Cast<HloDynamicSliceInstruction>(dynamic_slice)->index_shapes(),
1609 dynamic_slice->dynamic_slice_sizes()));
1622 parent_->evaluated_[dynamic_slice],
1625 absl::MakeConstSpan(dynamic_slice->operands()).subspan(1),
1630 parent_->evaluated_[dynamic_slice],
1633 absl::MakeConstSpan(dynamic_slice->operands()).subspan(1),
[all …]
|
| D | hlo_cost_analysis.h | 105 Status HandleDynamicSlice(const HloInstruction* dynamic_slice) override;
|
| D | hlo_verifier.cc | 845 Status ShapeVerifier::HandleDynamicSlice(HloInstruction* dynamic_slice) { in HandleDynamicSlice() argument
847 dynamic_slice, in HandleDynamicSlice()
849 dynamic_slice->operand(0)->shape(), in HandleDynamicSlice()
850 Cast<HloDynamicSliceInstruction>(dynamic_slice)->index_shapes(), in HandleDynamicSlice()
851 dynamic_slice->dynamic_slice_sizes())); in HandleDynamicSlice()
|
| D | hlo_verifier.h | 88 Status HandleDynamicSlice(HloInstruction* dynamic_slice) override;
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | xla_ops_test.py | 506 xla.dynamic_slice,
519 output = xla.dynamic_slice(
533 output = xla.dynamic_slice(
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/ |
| D | hlo_ops.cc | 1416 LogicalResult matchAndRewrite(DynamicSliceOp dynamic_slice, in matchAndRewrite()
1418 Value input = dynamic_slice.operand(); in matchAndRewrite()
1423 for (Value start : dynamic_slice.start_indices()) { in matchAndRewrite()
1433 auto loc = dynamic_slice.getLoc(); in matchAndRewrite()
1438 slice_start_indices, dynamic_slice.slice_sizes(), &rewriter); in matchAndRewrite()
1443 rewriter.replaceOp(dynamic_slice, {result}); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/python/ |
| D | xla.py | 320 dynamic_slice = gen_xla_ops.xla_dynamic_slice variable
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.h | 154 Status HandleDynamicSlice(HloInstruction* dynamic_slice) override;
|
| D | ir_emitter.cc | 1969 Status IrEmitter::HandleDynamicSlice(HloInstruction* dynamic_slice) { in HandleDynamicSlice() argument
1970 if (ShapeUtil::IsScalar(dynamic_slice->shape())) { in HandleDynamicSlice()
1971 TF_RETURN_IF_ERROR(EmitTargetAddressForOp(dynamic_slice)); in HandleDynamicSlice()
1972 return EmitMemcpy(*dynamic_slice->operand(0), *dynamic_slice); in HandleDynamicSlice()
1974 return DefaultAction(dynamic_slice); in HandleDynamicSlice()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | mhlo_to_lhlo_with_xla.cc | 1255 TF_ASSIGN_OR_RETURN(auto dynamic_slice, in EmitDynamicSliceOp()
1258 dynamic_slice.slice_sizesAttr( in EmitDynamicSliceOp()
1260 return dynamic_slice; in EmitDynamicSliceOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/ |
| D | lhlo_ops.td | 312 def LHLO_DynamicSliceOp: LHLO_Op<"dynamic_slice",
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/tests/ |
| D | ops.mlir | 765 // CHECK-LABEL: func @dynamic_slice
766 func @dynamic_slice(%arg0: tensor<3x4xi32>, %arg1: tensor<i64>, %arg2: tensor<i64>) -> tensor<1x4xi…
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/hlo_to_lhlo_with_xla/ |
| D | ops.mlir | 831 // CHECK: "lmhlo.dynamic_slice"(%arg0, %arg1, %arg2, %{{.*}}) {slice_sizes = dense<[1, 4]> : tensor…
|