Searched refs:dynamic_slice_shape (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_creation_utils.cc | 143 Shape dynamic_slice_shape, in MakeDynamicSliceHlo()
147 dynamic_slice_shape, operand, start_indices, slice_sizes)); in MakeDynamicSliceHlo()
170 Shape dynamic_slice_shape, in MakeDynamicSliceHlo()
174 dynamic_slice_shape, operand, scalar_start_indices, slice_sizes)); in MakeDynamicSliceHlo()
|
| D | algebraic_simplifier.cc | 1987 const Shape& dynamic_slice_shape = a->shape(); in OptimizeDotOfGather() local
1988 if (dynamic_slice_shape.dimensions(1 - a_contracting_dimension) != 1) { in OptimizeDotOfGather()
1996 dynamic_slice_shape.dimensions(a_contracting_dimension)) { in OptimizeDotOfGather()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | convolution_handler.cc | 356 Shape dynamic_slice_shape = lhs.hlo()->shape(); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS() local
441 dynamic_slice_shape.set_dimensions(lhs_dimension, in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
458 dynamic_slice_shape, pad, dynamic_slice_start_indices, in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
459 dynamic_slice_shape.dimensions())); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_instruction_fusion_test.cc | 340 Shape dynamic_slice_shape = ShapeUtil::MakeShape(F32, {4, 4}); in TEST_F() local
353 dynamic_slice_shape, reshape3, {param1, param2}, {4, 4})); in TEST_F()
355 dynamic_slice_shape, HloOpcode::kTanh, dynamic_slice4)); in TEST_F()
|