HomeSort by relevance Sort by last modified time
    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