Searched defs:dynamic_slice (Results 1 – 10 of 10) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | dot_operation_test.cc | 966 auto dynamic_slice = DynamicSlice(lhs_constant, {one, zero}, {1, 6}); in XLA_TEST_F() local
994 auto dynamic_slice = DynamicSlice(rhs_constant, {zero, one}, {6, 1}); in XLA_TEST_F() local
1024 auto dynamic_slice = DynamicSlice(lhs_constant, {zero, one}, {6, 1}); in XLA_TEST_F() local
1052 auto dynamic_slice = DynamicSlice(rhs_constant, {one, zero}, {1, 6}); in XLA_TEST_F() local
1085 auto dynamic_slice = DynamicSlice(lhs_constant, {zero, one}, {6, 1}); in XLA_TEST_F() local
1118 auto dynamic_slice = DynamicSlice(rhs_constant, {zero, one}, {6, 1}); in XLA_TEST_F() local
1143 auto dynamic_slice = DynamicSlice(lhs_constant, {one, zero}, {1, 6}); in XLA_TEST_F() local
1168 auto dynamic_slice = DynamicSlice(rhs_constant, {one, zero}, {1, 6}); in XLA_TEST_F() local
|
| H A D | value_inference_test.cc | 264 auto dynamic_slice = DynamicSlice(constant, {slice_start}, {1}); in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | while_loop_all_reduce_code_motion.cc | 50 std::optional<HloInstruction*> dynamic_slice; member
207 std::optional<HloInstruction*> dynamic_slice; in IsAllReduceMovable() member
433 const HloInstruction& dynamic_update_slice) -> bool { in IsAllReduceMovable()
|
| H A D | reduce_scatter_utils.h | 31 HloInstruction* dynamic_slice; member
|
| H A D | dfs_hlo_visitor_with_default.h | 183 Status HandleDynamicSlice(HloInstructionPtr dynamic_slice) override { in HandleDynamicSlice()
|
| H A D | hlo_cost_analysis.cc | 259 const HloInstruction* dynamic_slice) { in HandleDynamicSlice()
|
| H A D | hlo_verifier.cc | 1215 Status ShapeVerifier::HandleDynamicSlice(HloInstruction* dynamic_slice) { in HandleDynamicSlice()
|
| H A D | algebraic_simplifier.cc | 2339 HloDynamicSliceInstruction* dynamic_slice = in OptimizeDotOfGather() local
4806 HloInstruction* dynamic_slice) { in HandleDynamicSlice()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/python/ |
| H A D | xla.py | 366 dynamic_slice = gen_xla_ops.xla_dynamic_slice variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | ir_emitter.cc | 2080 Status IrEmitter::HandleDynamicSlice(HloInstruction* dynamic_slice) { in HandleDynamicSlice()
|