Home
last modified time | relevance | path

Searched refs:IsDenseArray (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dnccl_all_gather_thunk.cc43 return LayoutUtil::IsDenseArray(operand->shape()) && in CanImplement()
55 return LayoutUtil::IsDenseArray(shape) && in CanImplement()
Dnccl_all_to_all_thunk.cc47 return LayoutUtil::IsDenseArray(shape) && in CanImplement()
Dnccl_all_reduce_thunk.cc104 return LayoutUtil::IsDenseArray(shape) && in CanImplement()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_input_output_alias_config.cc203 TF_RET_CHECK(LayoutUtil::IsDenseArray(param_subshape)); in Verify()
204 TF_RET_CHECK(LayoutUtil::IsDenseArray(output_subshape)); in Verify()
Dpattern_matcher_test.cc77 EXPECT_TRUE(Match(&scalar_shape, match::Shape().IsDenseArray())); in TEST_F()
91 EXPECT_TRUE(Match(&array_shape, match::Shape().IsDenseArray())); in TEST_F()
Dhlo_verifier.cc1901 LayoutUtil::IsDenseArray(instruction->shape()) && in Postprocess()
1907 if (LayoutUtil::IsDenseArray(operand_shape) && in Postprocess()
Dpattern_matcher.h1025 constexpr auto IsDenseArray() const {
Dhlo_parser.cc3111 CHECK(LayoutUtil::IsDenseArray(shape)) << shape.ToString(true); in ParseNonTupleLiteral()
/external/tensorflow/tensorflow/compiler/xla/
Dlayout_util.h95 static bool IsDenseArray(const Shape& shape);
Dlayout_util.cc275 /* static */ bool LayoutUtil::IsDenseArray(const Shape& shape) { in IsDenseArray() function in xla::LayoutUtil
322 CHECK(IsDenseArray(shape)); in MinorToMajor()
Dliteral.cc453 CHECK(LayoutUtil::IsDenseArray(subshape())); in GetDynamicSize()
463 CHECK(LayoutUtil::IsDenseArray(subshape())); in SetDynamicSize()
872 CHECK(LayoutUtil::IsDenseArray(permuted_shape)); in Transpose()
975 CHECK(LayoutUtil::IsDenseArray(subshape)); in GetAsString()
1020 CHECK(LayoutUtil::IsDenseArray(shape())); in GetIntegralAsS64()
1047 CHECK(LayoutUtil::IsDenseArray(shape())); in GetAsDouble()
1107 CHECK(LayoutUtil::IsDenseArray(shape())); in SetIntegralAsS64()
1136 CHECK(LayoutUtil::IsDenseArray(shape())); in SetFromDouble()
1285 CHECK(LayoutUtil::IsDenseArray(subshape)); in ToStringHelper()
1630 LayoutUtil::IsDenseArray(subshape())) { in EqualElements()
Dshape.cc161 if (LayoutUtil::IsDenseArray(lhs)) { in operator ()()
Dliteral.h921 CHECK(LayoutUtil::IsDenseArray(subshape())); in Get()
929 CHECK(LayoutUtil::IsDenseArray(subshape())); in Set()
1062 TF_RET_CHECK(LayoutUtil::IsDenseArray(this_shape)); in PopulateInternal()
Dshape_util.cc417 CHECK(LayoutUtil::IsDenseArray(*shape)); in AppendMajorDimension()
772 CHECK(LayoutUtil::IsDenseArray(shape)) << shape.ShortDebugString(); in ByteSizeOfElements()
1099 CHECK(LayoutUtil::IsDenseArray(shape)); in PermuteDimensions()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_runtime.cc679 (num_buffers == 1 && xla::LayoutUtil::IsDenseArray(shape))); in __xla_cpu_runtime_AllReduce()