Searched refs:dynamic_slice_shape (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_creation_utils.cc | 141 Shape dynamic_slice_shape, in MakeDynamicSliceHlo()
145 dynamic_slice_shape, operand, scalar_start_indices, slice_sizes)); in MakeDynamicSliceHlo()
|
| D | algebraic_simplifier.cc | 1533 const Shape& dynamic_slice_shape = a->shape(); in OptimizeDotOfGather() local
1534 if (dynamic_slice_shape.dimensions(1 - a_contracting_dimension) != 1) { in OptimizeDotOfGather()
1542 dynamic_slice_shape.dimensions(a_contracting_dimension)) { in OptimizeDotOfGather()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_instruction_fusion_test.cc | 338 Shape dynamic_slice_shape = ShapeUtil::MakeShape(F32, {4, 4}); in TEST_F() local
351 dynamic_slice_shape, reshape3, {param1, param2}, {4, 4})); in TEST_F()
353 dynamic_slice_shape, HloOpcode::kTanh, dynamic_slice4)); in TEST_F()
|