Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dif_op.cc119 if (!xla::ShapeUtil::DynamicShapeIsCompatible(then_result.xla_output_shape, in ValidateShapes()
120 else_result.xla_output_shape) && in ValidateShapes()
121 !xla::ShapeUtil::DynamicShapeIsCompatible(else_result.xla_output_shape, in ValidateShapes()
122 then_result.xla_output_shape)) { in ValidateShapes()
129 xla::ShapeUtil::HumanString(then_result.xla_output_shape), " vs. ", in ValidateShapes()
130 xla::ShapeUtil::HumanString(else_result.xla_output_shape), in ValidateShapes()
138 xla::ShapeUtil::HumanString(then_result.xla_output_shape), " vs. ", in ValidateShapes()
139 xla::ShapeUtil::HumanString(else_result.xla_output_shape)); in ValidateShapes()
161 << xla::ShapeUtil::HumanString(then_result.xla_output_shape); in ValidateShapes()
Dwhile_op.cc185 body.xla_output_shape.tuple_shapes(i); in VerifyBodyInputAndOutputShapeMatch()
191 body.xla_output_shape.tuple_shapes(ctx->num_inputs()); in VerifyBodyInputAndOutputShapeMatch()
419 xla::Shape body_output_shape = body.xla_output_shape; in Compile()
461 << " -> " << xla::ShapeUtil::HumanString(body.xla_output_shape); in Compile()
463 << " -> " << xla::ShapeUtil::HumanString(cond.xla_output_shape); in Compile()
487 cond.xla_output_shape, in Compile()
490 cond.xla_output_shape, in Compile()
495 xla::ShapeUtil::HumanString(cond.xla_output_shape))); in Compile()
Dcase_op.cc234 branch_results[0].xla_output_shape); in Compile()
/external/tensorflow/tensorflow/core/tpu/kernels/xla/
Dhost_compute_ops.cc194 xla::Shape* xla_output_shape; in Compile() local
208 xla_output_shape = &inferred_xla_output_shape; in Compile()
212 xla_output_shape = &static_xla_output_shape_; in Compile()
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_compile_op_support.cc223 compilation_result.xla_output_shape, pos); in AddVariableUpdatesToCores()
286 compilation_result.xla_output_shape, i); in ComputeOutputShapesForEachCore()
328 compilation_result.xla_output_shape, device_assignment, in CreateHloModules()
Dtpu_util.cc61 instance.result_layout = &compilation_result.xla_output_shape; in BuildAotXlaComputationInstance()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_compiler_test.cc328 EXPECT_EQ(result.xla_output_shape, in TEST_F()
427 EXPECT_EQ(result.xla_output_shape, in TEST_F()
470 EXPECT_EQ(result.xla_output_shape, in TEST_F()
490 EXPECT_EQ(result.xla_output_shape, in TEST_F()
1189 result.xla_output_shape, in TEST_F()
1192 result.xla_output_shape); in TEST_F()
1234 result.xla_output_shape, in TEST_F()
1237 result.xla_output_shape); in TEST_F()
1622 EXPECT_TRUE(result.xla_output_shape.IsTuple()); in TEST_F()
1623 EXPECT_EQ(xla::ShapeUtil::TupleElementCount(result.xla_output_shape), 1); in TEST_F()
[all …]
Dxla_helpers.h143 xla::Shape xla_output_shape; member
Dxla_compiler.cc1429 &result->resource_updates, &result->xla_output_shape, in CompileGraph()
1435 << xla::ShapeUtil::HumanStringWithLayout(result->xla_output_shape); in CompileGraph()
/external/tensorflow/tensorflow/compiler/jit/
Dget_compiler_ir.cc61 build_options.set_result_layout(result.xla_output_shape); in BuildExecutable()
Dxla_compilation_cache.cc267 build_options.set_result_layout(result.xla_output_shape); in GetBuildOptions()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_executable.h155 absl::string_view module_name, Shape xla_output_shape,
Dgpu_executable.cc1009 absl::string_view module_name, Shape xla_output_shape, in GpuExecutable() argument
1017 output_shape_(xla_output_shape), in GpuExecutable()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dcompile_mlir_util.cc188 xla::Shape* xla_output_shape, std::vector<XlaOutputDescription>* outputs, in GetOutputInfo() argument
296 *xla_output_shape = xla::ShapeUtil::MakeTupleShape(shapes); in GetOutputInfo()
631 &compilation_result->xla_output_shape, &compilation_result->outputs, in PopulateResultIOInfo()
Dtf_xla_mlir_translate.cc106 << compilation_result.xla_output_shape.ToString() << '\n'; in PrintHloModuleText()