Home
last modified time | relevance | path

Searched defs:dynamic_slice (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Ddot_operation_test.cc925 auto dynamic_slice = DynamicSlice(lhs_constant, {one, zero}, {1, 6}); in XLA_TEST_F() local
953 auto dynamic_slice = DynamicSlice(rhs_constant, {zero, one}, {6, 1}); in XLA_TEST_F() local
983 auto dynamic_slice = DynamicSlice(lhs_constant, {zero, one}, {6, 1}); in XLA_TEST_F() local
1011 auto dynamic_slice = DynamicSlice(rhs_constant, {one, zero}, {1, 6}); in XLA_TEST_F() local
1044 auto dynamic_slice = DynamicSlice(lhs_constant, {zero, one}, {6, 1}); in XLA_TEST_F() local
1077 auto dynamic_slice = DynamicSlice(rhs_constant, {zero, one}, {6, 1}); in XLA_TEST_F() local
1102 auto dynamic_slice = DynamicSlice(lhs_constant, {one, zero}, {1, 6}); in XLA_TEST_F() local
1127 auto dynamic_slice = DynamicSlice(rhs_constant, {one, zero}, {1, 6}); in XLA_TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/service/
Ddfs_hlo_visitor_with_default.h151 Status HandleDynamicSlice(HloInstructionPtr dynamic_slice) override { in HandleDynamicSlice()
Dhlo_cost_analysis.cc197 const HloInstruction* dynamic_slice) { in HandleDynamicSlice()
Dhlo_verifier.cc575 Status ShapeVerifier::HandleDynamicSlice(HloInstruction* dynamic_slice) { in HandleDynamicSlice()
Dalgebraic_simplifier.cc1578 HloDynamicSliceInstruction* dynamic_slice = in OptimizeDotOfGather() local
2875 HloInstruction* dynamic_slice) { in HandleDynamicSlice()
Dhlo_evaluator_typed_visitor.h1456 Status HandleDynamicSlice(HloInstruction* dynamic_slice) override { in HandleDynamicSlice()
/external/tensorflow/tensorflow/compiler/jit/
Dpartially_decluster_pass_test.cc421 Output dynamic_slice = in TEST() local
/external/tensorflow/tensorflow/compiler/tf2xla/python/
Dxla.py305 dynamic_slice = gen_xla_ops.xla_dynamic_slice variable
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc2055 Status IrEmitter::HandleDynamicSlice(HloInstruction* dynamic_slice) { in HandleDynamicSlice()