Home
last modified time | relevance | path

Searched refs:has_layout (Results 1 – 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
Dlayout_util.cc177 if (shape.has_layout()) { in ValidateLayoutInShape()
186 if (!shape.has_layout()) { in ValidateLayoutInShape()
196 if (shape.has_layout()) { in ValidateLayoutInShape()
274 return shape.IsArray() && shape.has_layout() && IsDense(shape.layout()); in IsDenseArray()
294 return shape.IsArray() && shape.has_layout() && IsSparse(shape.layout()); in IsSparseArray()
315 return shape.has_layout() && shape.layout().format() != INVALID_FORMAT; in HasLayout()
392 if (src.has_layout()) { in CopyLayoutInternal()
Dlayout_util_test.cc89 EXPECT_TRUE(dst.has_layout()); in TEST_F()
92 EXPECT_FALSE(dst.has_layout()); in TEST_F()
112 EXPECT_TRUE(dst.has_layout()); in TEST_F()
115 EXPECT_FALSE(dst.has_layout()); in TEST_F()
270 EXPECT_TRUE(shape.tuple_shapes(0).has_layout()); in TEST_F()
271 EXPECT_TRUE(shape.tuple_shapes(2).tuple_shapes(1).has_layout()); in TEST_F()
276 EXPECT_FALSE(shape.tuple_shapes(0).has_layout()); in TEST_F()
277 EXPECT_FALSE(shape.tuple_shapes(2).tuple_shapes(1).has_layout()); in TEST_F()
Dshape.cc53 if (shape_proto.has_layout()) { in Shape()
72 if (has_layout()) { in ToProto()
Dshape.h124 bool has_layout() const { return layout_.format() != INVALID_FORMAT; } in has_layout() function
Dshape_util.cc107 if (!shape.has_layout()) { in MakeShapeWithLayoutInternal()
652 if (shape.has_layout()) { in ValidateShapeWithOptionalLayoutInternal()
969 if (shape.has_layout()) { in PermuteDimensions()
Dshape_util_test.cc276 ASSERT_TRUE(scalar_default_layout.has_layout()) in TEST()
281 ASSERT_TRUE(scalar_empty_min2maj.has_layout()) in TEST()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_layout_assignment_test.cc157 if (input_shape.has_layout()) { in TEST_F()
162 if (result_shape.has_layout()) { in TEST_F()
226 if (input_shape.has_layout()) { in TEST_F()
231 if (result_shape.has_layout()) { in TEST_F()
308 if (input_shape.has_layout()) { in TEST_F()
313 if (result_shape.has_layout()) { in TEST_F()
Dir_emitter_unnested.cc3767 CHECK(input_shape.has_layout()) << "LayoutAssignment or InstructionFusion " in EmitReductionToVector()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dshape_util.cc76 TF_ASSIGN_OR_RETURN(bool has_layout, MakeLayout(minor_to_major, &layout)); in AssignLayout()
77 if (!has_layout && layout_func) { in AssignLayout()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Ddot_op_emitter.cc754 CHECK(!lhs_shape.has_layout() || IsSimpleLayout(lhs_shape.layout())) in AreGemmShapes()
756 CHECK(!rhs_shape.has_layout() || IsSimpleLayout(rhs_shape.layout())) in AreGemmShapes()
758 CHECK(!output_shape.has_layout() || IsSimpleLayout(output_shape.layout())) in AreGemmShapes()
886 CHECK(shape.has_layout() && in CollapseFirstNDims()
/external/mesa3d/src/compiler/glsl/
Dglsl_parser.yy835 if (block->layout.has_layout() || block->layout.has_memory()) {
1961 $$.merge_qualifier(& @1, state, $2, false, $2.has_layout());
1977 $2.has_interpolation() || $2.has_layout())) {
1995 $2.has_layout() || $2.has_auxiliary_storage())) {
2699 block->layout.has_layout())) {
Dast.h764 bool has_layout() const;
Dast_type.cpp61 ast_type_qualifier::has_layout() const in has_layout() function in ast_type_qualifier
Dast_to_hir.cpp3797 if (qual->has_layout() && uses_deprecated_qualifier) { in apply_layout_qualifier_to_variable()
/external/tensorflow/tensorflow/compiler/xla/service/
Dlayout_assignment.cc1493 TF_RET_CHECK(!subshape.has_layout()); in InferArrayLayout()
1508 if (!source_buffer->shape().has_layout()) { in InferArrayLayout()
1635 if (subshape->has_layout() || !subshape->IsArray()) { in AssignLayouts()
Dpattern_matcher.h970 << (shape->has_layout() ? ShapeUtil::HumanStringWithLayout(*shape)
985 << (shape->has_layout() ? ShapeUtil::HumanStringWithLayout(*shape)
Dhlo_instructions.cc1053 if (!mutable_array_subshape->has_layout() || in RelayoutConstant()