| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | xla_select_and_scatter_op.cc | 80 xla::ShapeUtil::Compatible(select.xla_output_shape, in Compile()
85 xla::ShapeUtil::HumanString(select.xla_output_shape))); in Compile()
103 xla::ShapeUtil::Compatible(scatter.xla_output_shape, scalar_shape), in Compile()
107 xla::ShapeUtil::HumanString(scatter.xla_output_shape))); in Compile()
|
| D | if_op.cc | 115 if (!xla::ShapeUtil::Compatible(then_result.xla_output_shape, in ValidateShapes()
116 else_result.xla_output_shape)) { in ValidateShapes()
119 xla::ShapeUtil::HumanString(then_result.xla_output_shape), " vs. ", in ValidateShapes()
120 xla::ShapeUtil::HumanString(else_result.xla_output_shape)); in ValidateShapes()
142 << xla::ShapeUtil::HumanString(then_result.xla_output_shape); in ValidateShapes()
|
| D | case_op.cc | 230 xla::ShapeUtil::Compatible(branch_results[0].xla_output_shape, in Compile()
231 branch_results[j].xla_output_shape), in Compile()
234 xla::ShapeUtil::HumanString(branch_results[0].xla_output_shape), in Compile()
236 xla::ShapeUtil::HumanString(branch_results[j].xla_output_shape))); in Compile()
243 branch_results[0].xla_output_shape); in Compile()
|
| D | while_op.cc | 180 body.xla_output_shape.tuple_shapes(i); in VerifyBodyInputAndOutputShapeMatch()
186 body.xla_output_shape.tuple_shapes(ctx->num_inputs()); in VerifyBodyInputAndOutputShapeMatch()
407 xla::Shape body_output_shape = body.xla_output_shape; in Compile()
449 << " -> " << xla::ShapeUtil::HumanString(body.xla_output_shape); in Compile()
451 << " -> " << xla::ShapeUtil::HumanString(cond.xla_output_shape); in Compile()
475 cond.xla_output_shape, in Compile()
478 cond.xla_output_shape, in Compile()
483 xla::ShapeUtil::HumanString(cond.xla_output_shape))); in Compile()
|
| D | reduce_window_op.cc | 96 xla::ShapeUtil::Compatible(reducer.xla_output_shape, scalar_shape), in Compile()
100 xla::ShapeUtil::HumanString(reducer.xla_output_shape))); in Compile()
|
| D | sort_ops.cc | 106 xla::ShapeUtil::Compatible(comparator.xla_output_shape, in Compile()
111 " got ", xla::ShapeUtil::HumanString(comparator.xla_output_shape))); in Compile()
|
| D | xla_reduce_op.cc | 98 xla::ShapeUtil::Compatible(reducer.xla_output_shape, expected_shape), in Compile()
102 xla::ShapeUtil::HumanString(reducer.xla_output_shape))); in Compile()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_helpers.h | 149 xla::Shape xla_output_shape; member
|
| D | xla_compiler_test.cc | 325 EXPECT_EQ(result.xla_output_shape, in TEST_F()
419 EXPECT_EQ(result.xla_output_shape, in TEST_F()
462 EXPECT_EQ(result.xla_output_shape, in TEST_F()
482 EXPECT_EQ(result.xla_output_shape, in TEST_F()
1116 result.xla_output_shape, in TEST_F()
1119 result.xla_output_shape); in TEST_F()
1158 result.xla_output_shape, in TEST_F()
1161 result.xla_output_shape); in TEST_F()
1543 EXPECT_TRUE(result.xla_output_shape.IsTuple()); in TEST_F()
1544 EXPECT_EQ(xla::ShapeUtil::TupleElementCount(result.xla_output_shape), 1); in TEST_F()
[all …]
|
| D | xla_compiler.cc | 1394 &result->resource_updates, &result->xla_output_shape, in CompileGraph()
1400 << xla::ShapeUtil::HumanStringWithLayout(result->xla_output_shape); in CompileGraph()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_util.cc | 61 instance.result_layout = &compilation_result.xla_output_shape; in BuildAotXlaComputationInstance()
|
| D | tpu_compile_op_support.cc | 221 compilation_result.xla_output_shape, pos); in AddVariableUpdatesToCores()
281 compilation_result.xla_output_shape, i); in ComputeOutputShapesForEachCore()
323 compilation_result.xla_output_shape, device_assignment, in CreateHloModules()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/xla/ |
| D | host_compute_ops.cc | 180 xla::Shape* xla_output_shape; in Compile() local
194 xla_output_shape = &inferred_xla_output_shape; in Compile()
198 xla_output_shape = &static_xla_output_shape_; in Compile()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | compile_mlir_util.cc | 139 xla::Shape* xla_output_shape, std::vector<XlaOutputDescription>* outputs, in GetOutputInfo() argument
198 *xla_output_shape = xla::ShapeUtil::MakeTupleShape(shapes); in GetOutputInfo()
463 &compilation_result->xla_output_shape, &compilation_result->outputs, in PopulateResultIOInfo()
|
| D | tf_xla_mlir_translate.cc | 92 << compilation_result.xla_output_shape.ToString() << '\n'; in PrintHloModuleText()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_compilation_cache.cc | 172 build_options.set_result_layout(result.xla_output_shape); in BuildExecutable()
|