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

  /external/tensorflow/tensorflow/compiler/xla/
layout_util.h 72 // HasLayout will return false for the shape.
127 static bool HasLayout(const Shape& shape);
130 static bool HasLayout(const ProgramShape& program_shape);
shape_layout.cc 61 bool ShapeLayout::LayoutIsSet() const { return LayoutUtil::HasLayout(shape_); }
layout_util.cc 266 if (ShapeUtil::IsTuple(shape) || !HasLayout(shape) ||
311 /* static */ bool LayoutUtil::HasLayout(const Shape& shape) {
315 [](const Shape& s) { return HasLayout(s); });
322 /* static */ bool LayoutUtil::HasLayout(const ProgramShape& program_shape) {
324 if (!LayoutUtil::HasLayout(parameter_shape)) {
328 return LayoutUtil::HasLayout(program_shape.result());
shape_util.cc 475 if (LayoutUtil::HasLayout(shape)) {
796 if (LayoutUtil::HasLayout(shape)) {
    [all...]
layout_util_test.cc 228 EXPECT_TRUE(LayoutUtil::HasLayout(shape));
234 EXPECT_FALSE(LayoutUtil::HasLayout(shape));
literal_util.cc 91 CHECK(LayoutUtil::HasLayout(shape));
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/
transfer_manager.cc 186 if (!LayoutUtil::HasLayout(on_host_shape)) {
193 TF_RET_CHECK(LayoutUtil::HasLayout(on_device_shape));
hlo_tfgraph_builder.cc 161 if (LayoutUtil::HasLayout(instruction->shape())) {
layout_assignment.cc 65 TF_RET_CHECK(LayoutUtil::HasLayout(shape_with_layout));
118 TF_RET_CHECK(LayoutUtil::HasLayout(operand->shape()));
    [all...]
layout_assignment_test.cc 177 EXPECT_FALSE(LayoutUtil::HasLayout(
213 EXPECT_TRUE(LayoutUtil::HasLayout(tuple->shape()));
    [all...]
algebraic_simplifier.cc     [all...]
service.cc 227 if (!LayoutUtil::HasLayout(shape_with_layout)) {
    [all...]
user_computation.cc     [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
ir_array.cc 38 CHECK(LayoutUtil::HasLayout(shape))
74 CHECK(LayoutUtil::HasLayout(shape))
85 CHECK(LayoutUtil::HasLayout(shape));

Completed in 406 milliseconds