Home
last modified time | relevance | path

Searched defs:new_tile_shape (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_sharding_util.cc1287 std::vector<int64_t> new_tile_shape( in PartiallyReplicateTiledShardingOnDims() local
1338 std::vector<int64_t> new_tile_shape(data_rank, 1); in ReplicateAllDataDims() local
1355 std::vector<int64_t> new_tile_shape; in RemoveShapeDimensions() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/
H A Dxla_builder.cc5024 std::vector<int64_t> new_tile_shape( in GetManualSharding() local