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

  /external/tensorflow/tensorflow/compiler/xla/
layout_util.cc 177 if (shape.has_layout()) {
186 if (!shape.has_layout()) {
196 if (shape.has_layout()) {
274 return shape.IsArray() && shape.has_layout() && IsDense(shape.layout());
294 return shape.IsArray() && shape.has_layout() && IsSparse(shape.layout());
315 return shape.has_layout() && shape.layout().format() != INVALID_FORMAT;
392 if (src.has_layout()) {
layout_util_test.cc 89 EXPECT_TRUE(dst.has_layout());
92 EXPECT_FALSE(dst.has_layout());
112 EXPECT_TRUE(dst.has_layout());
115 EXPECT_FALSE(dst.has_layout());
270 EXPECT_TRUE(shape.tuple_shapes(0).has_layout());
271 EXPECT_TRUE(shape.tuple_shapes(2).tuple_shapes(1).has_layout());
276 EXPECT_FALSE(shape.tuple_shapes(0).has_layout());
277 EXPECT_FALSE(shape.tuple_shapes(2).tuple_shapes(1).has_layout());
shape.cc 53 if (shape_proto.has_layout()) {
72 if (has_layout()) {
shape.h 124 bool has_layout() const { return layout_.format() != INVALID_FORMAT; } function in class:xla::Shape
shape_util.cc 107 if (!shape.has_layout()) {
652 if (shape.has_layout()) {
    [all...]
shape_util_test.cc 276 ASSERT_TRUE(scalar_default_layout.has_layout())
281 ASSERT_TRUE(scalar_empty_min2maj.has_layout())
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
gpu_layout_assignment_test.cc 157 if (input_shape.has_layout()) {
162 if (result_shape.has_layout()) {
226 if (input_shape.has_layout()) {
231 if (result_shape.has_layout()) {
308 if (input_shape.has_layout()) {
313 if (result_shape.has_layout()) {
ir_emitter_unnested.cc     [all...]
  /external/tensorflow/tensorflow/compiler/tf2xla/
shape_util.cc 76 TF_ASSIGN_OR_RETURN(bool has_layout, MakeLayout(minor_to_major, &layout));
77 if (!has_layout && layout_func) {
  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
dot_op_emitter.cc 754 CHECK(!lhs_shape.has_layout() || IsSimpleLayout(lhs_shape.layout()))
756 CHECK(!rhs_shape.has_layout() || IsSimpleLayout(rhs_shape.layout()))
758 CHECK(!output_shape.has_layout() || IsSimpleLayout(output_shape.layout()))
    [all...]
  /external/mesa3d/src/compiler/glsl/
ast.h 764 bool has_layout() const;
    [all...]
ast_type.cpp 61 ast_type_qualifier::has_layout() const function in class:ast_type_qualifier
    [all...]
ast_to_hir.cpp     [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/
layout_assignment.cc     [all...]
pattern_matcher.h     [all...]
hlo_instructions.cc     [all...]

Completed in 1436 milliseconds