Home
last modified time | relevance | path

Searched refs:xla_output_shape (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dxla_select_and_scatter_op.cc81 xla::ShapeUtil::Compatible(select.xla_output_shape, in Compile()
86 xla::ShapeUtil::HumanString(select.xla_output_shape))); in Compile()
104 xla::ShapeUtil::Compatible(scatter.xla_output_shape, scalar_shape), in Compile()
108 xla::ShapeUtil::HumanString(scatter.xla_output_shape))); in Compile()
Dif_op.cc170 xla::ShapeUtil::Compatible(then_result.xla_output_shape, in Compile()
171 else_result.xla_output_shape), in Compile()
174 xla::ShapeUtil::HumanString(then_result.xla_output_shape), " vs. ", in Compile()
175 xla::ShapeUtil::HumanString(else_result.xla_output_shape))); in Compile()
179 << xla::ShapeUtil::HumanString(then_result.xla_output_shape); in Compile()
Dwhile_op.cc162 body.xla_output_shape.tuple_shapes(i); in VerifyBodyInputAndOutputShapeMatch()
168 body.xla_output_shape.tuple_shapes(ctx->num_inputs()); in VerifyBodyInputAndOutputShapeMatch()
386 xla::Shape body_output_shape = body.xla_output_shape; in Compile()
429 << " -> " << xla::ShapeUtil::HumanString(body.xla_output_shape); in Compile()
431 << " -> " << xla::ShapeUtil::HumanString(cond.xla_output_shape); in Compile()
455 cond.xla_output_shape, in Compile()
458 cond.xla_output_shape, in Compile()
463 xla::ShapeUtil::HumanString(cond.xla_output_shape))); in Compile()
Dcase_op.cc174 xla::ShapeUtil::Compatible(branch_results[0].xla_output_shape, in Compile()
175 branch_results[j].xla_output_shape), in Compile()
178 xla::ShapeUtil::HumanString(branch_results[0].xla_output_shape), in Compile()
180 xla::ShapeUtil::HumanString(branch_results[j].xla_output_shape))); in Compile()
187 branch_results[0].xla_output_shape); in Compile()
Dxla_reduce_op.cc80 xla::ShapeUtil::Compatible(reducer.xla_output_shape, scalar_shape), in Compile()
84 xla::ShapeUtil::HumanString(reducer.xla_output_shape))); in Compile()
Dreduce_window_op.cc97 xla::ShapeUtil::Compatible(reducer.xla_output_shape, scalar_shape), in Compile()
101 xla::ShapeUtil::HumanString(reducer.xla_output_shape))); in Compile()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_compiler_test.cc328 EXPECT_EQ(result.xla_output_shape, in TEST_F()
370 EXPECT_EQ(result.xla_output_shape, in TEST_F()
1051 result.xla_output_shape, in TEST_F()
1091 result.xla_output_shape, in TEST_F()
1475 EXPECT_TRUE(result.xla_output_shape.IsTuple()); in TEST_F()
1476 EXPECT_EQ(xla::ShapeUtil::TupleElementCount(result.xla_output_shape), 1); in TEST_F()
1494 EXPECT_TRUE(result.xla_output_shape.IsTuple()); in TEST_F()
1495 EXPECT_EQ(xla::ShapeUtil::TupleElementCount(result.xla_output_shape), 2); in TEST_F()
1496 EXPECT_TRUE(xla::ShapeUtil::GetTupleElementShape(result.xla_output_shape, 1) in TEST_F()
Dxla_compiler.h259 xla::Shape xla_output_shape; member
Dxla_compiler.cc1169 &result->resource_updates, &result->xla_output_shape)); in CompileGraph()
1174 << xla::ShapeUtil::HumanStringWithLayout(result->xla_output_shape); in CompileGraph()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_compilation_cache.cc155 build_options.set_result_layout(result.xla_output_shape); in BuildExecutable()