OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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