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

  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
cpu_hlo_support_checker.cc 32 if (LayoutUtil::IsSparseArray(subshape)) {
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
gpu_hlo_support_checker.cc 32 if (LayoutUtil::IsSparseArray(subshape)) {
  /external/tensorflow/tensorflow/compiler/xla/
layout_util.h 113 static bool IsSparseArray(const Shape& shape);
literal.h 352 // LayoutUtil::IsSparseArray(shape()) is true.
370 return LayoutUtil::IsSparseArray(subshape())
    [all...]
shape_util.cc 577 if (LayoutUtil::IsSparseArray(shape)) {
604 if (LayoutUtil::IsSparseArray(shape)) {
616 CHECK(LayoutUtil::IsSparseArray(shape));
661 if (LayoutUtil::IsSparseArray(shape) && shape.rank() == 0) {
690 if (shape_has_valid_layout && LayoutUtil::IsSparseArray(shape)) {
    [all...]
literal.cc 124 if (LayoutUtil::IsSparseArray(shape)) {
836 CHECK(LayoutUtil::IsSparseArray(subshape));
    [all...]
layout_util.cc 293 /* static */ bool LayoutUtil::IsSparseArray(const Shape& shape) {
  /external/tensorflow/tensorflow/compiler/xla/service/
pattern_matcher_test.cc 89 EXPECT_FALSE(Match(&array_shape, match::Shape().IsSparseArray()));
112 EXPECT_TRUE(Match(&array_shape, match::Shape().IsSparseArray()));
    [all...]
hlo_verifier.cc 34 if (LayoutUtil::IsSparseArray(subshape)) {
    [all...]
pattern_matcher.h 76 // - IsDenseArray/IsSparseArray
    [all...]
hlo_parser.cc     [all...]

Completed in 821 milliseconds