OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:HasLayout
(Results
1 - 23
of
23
) sorted by null
/external/tensorflow/tensorflow/compiler/xla/service/
zero_sized_hlo_elimination.cc
43
if (!LayoutUtil::
HasLayout
(shape)) {
optimize_input_output_buffer_alias.cc
43
TF_RET_CHECK(LayoutUtil::
HasLayout
(input_shape));
44
TF_RET_CHECK(LayoutUtil::
HasLayout
(output_shape));
transfer_manager.cc
312
if (!LayoutUtil::
HasLayout
(on_host_shape)) {
318
TF_RET_CHECK(LayoutUtil::
HasLayout
(on_device_shape));
layout_assignment.cc
406
TF_RET_CHECK(LayoutUtil::
HasLayout
(data_shape));
[
all
...]
hlo_verifier.cc
552
TF_RET_CHECK(LayoutUtil::
HasLayout
(custom_call->shape()));
562
TF_RET_CHECK(LayoutUtil::
HasLayout
(operand_shape_with_layout));
[
all
...]
layout_assignment_test.cc
206
EXPECT_FALSE(LayoutUtil::
HasLayout
(
242
EXPECT_TRUE(LayoutUtil::
HasLayout
(tuple->shape()));
632
"LayoutUtil::
HasLayout
");
649
"LayoutUtil::
HasLayout
");
[
all
...]
algebraic_simplifier.cc
[
all
...]
hlo_cost_analysis.cc
126
if (!LayoutUtil::
HasLayout
(shape)) {
pattern_matcher.h
868
return LayoutUtil::
HasLayout
(*shape) &&
873
if (!LayoutUtil::
HasLayout
(*shape)) {
[
all
...]
service.cc
[
all
...]
hlo_evaluator.cc
[
all
...]
hlo_parser.cc
[
all
...]
/external/tensorflow/tensorflow/compiler/xla/
layout_util.h
87
//
HasLayout
will return false for the shape.
124
static bool
HasLayout
(const Shape& shape);
127
static bool
HasLayout
(const ProgramShape& program_shape);
shape_layout.cc
61
bool ShapeLayout::LayoutIsSet() const { return LayoutUtil::
HasLayout
(shape_); }
layout_util.cc
306
/* static */ bool LayoutUtil::
HasLayout
(const Shape& shape) {
310
[](const Shape& s) { return
HasLayout
(s); });
318
/* static */ bool LayoutUtil::
HasLayout
(const ProgramShape& program_shape) {
320
if (!LayoutUtil::
HasLayout
(parameter_shape)) {
324
return LayoutUtil::
HasLayout
(program_shape.result());
layout_util_test.cc
269
EXPECT_TRUE(LayoutUtil::
HasLayout
(shape));
275
EXPECT_FALSE(LayoutUtil::
HasLayout
(shape));
284
EXPECT_TRUE(LayoutUtil::
HasLayout
(shape));
286
EXPECT_TRUE(LayoutUtil::
HasLayout
(shape));
literal.cc
148
CHECK(LayoutUtil::
HasLayout
(*shape_));
299
if (!LayoutUtil::
HasLayout
(shape)) {
[
all
...]
shape_util.cc
471
} else if (shape.IsArray() && LayoutUtil::
HasLayout
(shape)) {
686
bool shape_has_valid_layout = LayoutUtil::
HasLayout
(shape) &&
[
all
...]
shape.cc
103
if (LayoutUtil::
HasLayout
(*this)) {
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
ir_array.cc
78
CHECK(LayoutUtil::
HasLayout
(shape))
96
CHECK(LayoutUtil::
HasLayout
(shape))
172
if (linear() != nullptr && LayoutUtil::
HasLayout
(input_shape) &&
173
LayoutUtil::
HasLayout
(output_shape) &&
208
if (linear() != nullptr && LayoutUtil::
HasLayout
(operand_shape) &&
209
LayoutUtil::
HasLayout
(shape) &&
220
CHECK(LayoutUtil::
HasLayout
(shape) && LayoutUtil::
HasLayout
(operand_shape));
259
if (linear_ == nullptr || !LayoutUtil::
HasLayout
(operand_shape) ||
260
!LayoutUtil::
HasLayout
(shape))
[
all
...]
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
ir_emission_utils.cc
30
CHECK(!LayoutUtil::
HasLayout
(shape) || LayoutUtil::IsDense(shape.layout()));
/external/tensorflow/tensorflow/compiler/tf2xla/
shape_util.cc
37
} else if (xla::LayoutUtil::
HasLayout
(shape)) {
/external/tensorflow/tensorflow/compiler/xla/client/
xla_builder.cc
[
all
...]
Completed in 1080 milliseconds