HomeSort by relevance Sort by last modified time
    Searched refs:tile_shape_ (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/
hlo_sharding.h 176 protobuf_util::ProtobufEquals(tile_shape_, other.tile_shape_) &&
197 for (uint32 v : tile_shape_.dimensions()) {
205 const Shape& tile_shape() const { return tile_shape_; }
215 tile_shape_(),
221 tile_shape_(),
227 tile_shape_(tile_shape),
244 Shape tile_shape_; variable
hlo_sharding.cc 60 return StrCat("{", ShapeUtil::HumanString(tile_shape_), " ",
77 CHECK(!ShapeUtil::IsTuple(tile_shape_));
97 CHECK_EQ(ShapeUtil::Rank(tile_shape_), tile_assignment_.dimensions().size());
106 // Index will always be all zeroes if we're maximal, and tile_shape_ is not
111 index[i] *= tile_shape_.dimensions(i);
122 index[i] = (index[i] + 1) * tile_shape_.dimensions(i);
247 if (ShapeUtil::Rank(shape) != ShapeUtil::Rank(tile_shape_)) {
256 if (ShapeUtil::Equal(shape, tile_shape_)) {
265 auto tile_dim = tile_shape_.dimensions(i);
278 CeilOfRatio(shape.dimensions(i), tile_shape_.dimensions(i))
    [all...]

Completed in 557 milliseconds