| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | dot_operation_test.cc | 925 auto dynamic_slice = DynamicSlice(lhs_constant, {one, zero}, {1, 6}); in XLA_TEST_F() local
930 DotGeneral(dynamic_slice, rhs_constant, dot_dnums); in XLA_TEST_F()
953 auto dynamic_slice = DynamicSlice(rhs_constant, {zero, one}, {6, 1}); in XLA_TEST_F() local
958 DotGeneral(lhs_constant, dynamic_slice, dot_dnums); in XLA_TEST_F()
983 auto dynamic_slice = DynamicSlice(lhs_constant, {zero, one}, {6, 1}); in XLA_TEST_F() local
988 DotGeneral(dynamic_slice, rhs_constant, dot_dnums); in XLA_TEST_F()
1011 auto dynamic_slice = DynamicSlice(rhs_constant, {one, zero}, {1, 6}); in XLA_TEST_F() local
1016 DotGeneral(lhs_constant, dynamic_slice, dot_dnums); in XLA_TEST_F()
1044 auto dynamic_slice = DynamicSlice(lhs_constant, {zero, one}, {6, 1}); in XLA_TEST_F() local
1049 DotGeneral(dynamic_slice, rhs_constant, dot_dnums); in XLA_TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | xla_ops_test.py | 303 xla.dynamic_slice,
315 output = xla.dynamic_slice(
328 output = xla.dynamic_slice(
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dfs_hlo_visitor_with_default.h | 151 Status HandleDynamicSlice(HloInstructionPtr dynamic_slice) override { in HandleDynamicSlice() argument
152 return DefaultAction(dynamic_slice); in HandleDynamicSlice()
|
| D | hlo_evaluator_typed_visitor.h | 1456 Status HandleDynamicSlice(HloInstruction* dynamic_slice) override { in HandleDynamicSlice() argument
1457 auto operand = dynamic_slice->operand(0); in HandleDynamicSlice()
1458 auto start_indices = dynamic_slice->operand(1); in HandleDynamicSlice()
1459 auto result_shape = dynamic_slice->shape(); in HandleDynamicSlice()
1464 Cast<HloDynamicSliceInstruction>(dynamic_slice)->index_shapes(), in HandleDynamicSlice()
1465 dynamic_slice->dynamic_slice_sizes())); in HandleDynamicSlice()
1478 parent_->evaluated_[dynamic_slice], in HandleDynamicSlice()
1481 absl::MakeConstSpan(dynamic_slice->operands()).subspan(1), in HandleDynamicSlice()
1486 parent_->evaluated_[dynamic_slice], in HandleDynamicSlice()
1489 absl::MakeConstSpan(dynamic_slice->operands()).subspan(1), in HandleDynamicSlice()
[all …]
|
| D | hlo_verifier.cc | 575 Status ShapeVerifier::HandleDynamicSlice(HloInstruction* dynamic_slice) { in HandleDynamicSlice() argument
577 dynamic_slice, in HandleDynamicSlice()
579 dynamic_slice->operand(0)->shape(), in HandleDynamicSlice()
580 Cast<HloDynamicSliceInstruction>(dynamic_slice)->index_shapes(), in HandleDynamicSlice()
581 dynamic_slice->dynamic_slice_sizes())); in HandleDynamicSlice()
|
| D | algebraic_simplifier.cc | 234 Status HandleDynamicSlice(HloInstruction* dynamic_slice) override;
1578 HloDynamicSliceInstruction* dynamic_slice = in OptimizeDotOfGather() local
1606 auto indices_type = dynamic_slice->operand(1)->shape().element_type(); in OptimizeDotOfGather()
1610 dynamic_slice->mutable_operand(1 + index_of_non_zero_start); in OptimizeDotOfGather()
1612 dynamic_slice->mutable_operand(1 + index_of_zero_start); in OptimizeDotOfGather()
2875 HloInstruction* dynamic_slice) { in HandleDynamicSlice() argument
2876 auto operand = dynamic_slice->mutable_operand(0); in HandleDynamicSlice()
2877 if (ShapeUtil::IsScalar(dynamic_slice->shape())) { in HandleDynamicSlice()
2878 return ReplaceInstruction(dynamic_slice, operand); in HandleDynamicSlice()
2882 if (SameShape(operand, dynamic_slice)) { in HandleDynamicSlice()
[all …]
|
| D | hlo_cost_analysis.h | 96 Status HandleDynamicSlice(const HloInstruction* dynamic_slice) override;
|
| D | hlo_cost_analysis.cc | 197 const HloInstruction* dynamic_slice) { in HandleDynamicSlice() argument
199 GetShapeSize(dynamic_slice->shape()) * 2; in HandleDynamicSlice()
|
| D | hlo_verifier.h | 79 Status HandleDynamicSlice(HloInstruction* dynamic_slice) override;
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | partially_decluster_pass_test.cc | 421 Output dynamic_slice = in TEST() local
428 ops::Reshape(s.WithOpName("reshape"), reshape_input, dynamic_slice); in TEST()
430 AddToCluster({dynamic_slice.node(), reshape.node()}, "cluster_0"); in TEST()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/python/ |
| D | xla.py | 305 dynamic_slice = gen_xla_ops.xla_dynamic_slice variable
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.h | 168 Status HandleDynamicSlice(HloInstruction* dynamic_slice) override;
|
| D | ir_emitter.cc | 2055 Status IrEmitter::HandleDynamicSlice(HloInstruction* dynamic_slice) { in HandleDynamicSlice() argument
2056 if (ShapeUtil::IsScalar(dynamic_slice->shape())) { in HandleDynamicSlice()
2057 TF_RETURN_IF_ERROR(EmitTargetAddressForOp(dynamic_slice)); in HandleDynamicSlice()
2058 return EmitMemcpy(*dynamic_slice->operand(0), *dynamic_slice); in HandleDynamicSlice()
2060 return DefaultAction(dynamic_slice); in HandleDynamicSlice()
|