| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_computation.h | 29 class XlaComputation {
31 XlaComputation() : unique_id_(-1) {} in XlaComputation() function
32 XlaComputation(HloModuleProto proto) in XlaComputation() function
35 ~XlaComputation() {} in ~XlaComputation()
37 XlaComputation(const XlaComputation&) = delete;
38 XlaComputation& operator=(const XlaComputation&) = delete;
40 XlaComputation(XlaComputation&& from) = default;
42 XlaComputation& operator=(XlaComputation&& from) = default;
58 XlaComputation(const int64 unique_id) : unique_id_(unique_id) {} in XlaComputation() function
|
| D | client.h | 62 const XlaComputation& computation,
90 const XlaComputation& computation,
101 const XlaComputation& computation;
106 XlaComputationInstance(const XlaComputation& computation, in XlaComputationInstance()
171 const XlaComputation& computation,
192 const XlaComputation& computation,
204 const XlaComputation& computation,
214 const XlaComputation& computation);
225 StatusOr<XlaComputation> LoadSnapshot(const HloSnapshot& module);
232 StatusOr<string> ExecutionStatsAsString(const XlaComputation& computation,
|
| D | xla_builder.h | 58 const XlaComputation& fused_computation);
275 StatusOr<XlaComputation> Build(bool remove_dynamic_dimensions = false);
279 StatusOr<XlaComputation> Build(XlaOp root,
290 XlaComputation BuildAndNoteError();
296 StatusOr<XlaComputation> BuildConstantSubGraph(
322 StatusOr<XlaComputation> BuildDynamicInferenceGraph(XlaOp root_op);
437 StatusOr<XlaComputation> Build(int64 root_id, bool remove_dynamic_dimensions);
649 XlaOp Call(const XlaComputation& computation,
675 const XlaComputation& computation, const Shape& shape_with_layout,
684 const XlaComputation& computation,
[all …]
|
| D | client.cc | 167 const XlaComputation& computation, absl::Span<GlobalData* const> arguments, in ExecuteAndTransfer()
184 StatusOr<Literal> Client::ComputeConstant(const XlaComputation& computation, in ComputeConstant()
212 StatusOr<XlaComputation> Client::LoadSnapshot(const HloSnapshot& module) { in LoadSnapshot()
214 return XlaComputation(module.hlo().hlo_module()); in LoadSnapshot()
218 const XlaComputation& computation, absl::Span<const Shape> argument_shapes, in Compile()
278 const XlaComputation& computation, absl::Span<GlobalData* const> arguments, in Execute()
424 const XlaComputation& computation, in GetComputationStats()
445 const XlaComputation& computation) { in GetComputationShape()
467 const XlaComputation& computation, const ExecutionProfile& profile) { in ExecutionStatsAsString()
|
| D | xla_computation.cc | 26 StatusOr<ProgramShape> XlaComputation::GetProgramShape() const { in GetProgramShape()
31 StatusOr<std::unique_ptr<HloSnapshot>> XlaComputation::Snapshot() const { in Snapshot()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | arithmetic.h | 30 XlaComputation CreateScalarComputation(const string& name, PrimitiveType type,
35 XlaComputation CreateScalarAddComputation(PrimitiveType type,
39 XlaComputation CreateScalarMultiplyComputation(PrimitiveType type,
43 XlaComputation CreateScalarGeComputation(PrimitiveType type,
47 XlaComputation CreateScalarMaxComputation(PrimitiveType type,
51 XlaComputation CreateScalarMinComputation(PrimitiveType type,
55 XlaComputation CreateScalarAndComputation(PrimitiveType type,
59 XlaComputation CreateScalarOrComputation(PrimitiveType type,
71 XlaComputation CreateScalarIdentityWithZeroComputation(PrimitiveType type,
|
| D | arithmetic.cc | 31 XlaComputation CreateScalarComputation(const string& name, PrimitiveType type, in CreateScalarComputation()
49 XlaComputation CreateScalarAddComputation(PrimitiveType type, in CreateScalarAddComputation()
55 XlaComputation CreateScalarMultiplyComputation(PrimitiveType type, in CreateScalarMultiplyComputation()
61 XlaComputation CreateScalarGeComputation(PrimitiveType type, in CreateScalarGeComputation()
67 XlaComputation CreateScalarMaxComputation(PrimitiveType type, in CreateScalarMaxComputation()
73 XlaComputation CreateScalarMinComputation(PrimitiveType type, in CreateScalarMinComputation()
79 XlaComputation CreateScalarAndComputation(PrimitiveType type, in CreateScalarAndComputation()
85 XlaComputation CreateScalarOrComputation(PrimitiveType type, in CreateScalarOrComputation()
91 XlaComputation CreateScalarIdentityWithZeroComputation(PrimitiveType type, in CreateScalarIdentityWithZeroComputation()
93 XlaComputation reducer = in CreateScalarIdentityWithZeroComputation()
[all …]
|
| D | comparators.h | 33 XlaComputation CreateScalarLtComputation(
42 XlaComputation CreateScalarGtComputation(
48 XlaComputation CreateScalarComparisonComputation(
|
| D | comparators.cc | 37 XlaComputation CreateScalarComparisonComputation( in CreateScalarComparisonComputation()
48 XlaComputation CreateScalarComparisonComputation( in CreateScalarComparisonComputation()
114 XlaComputation CreateScalarLtComputation( in CreateScalarLtComputation()
121 XlaComputation CreateScalarGtComputation( in CreateScalarGtComputation()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | call_test.cc | 36 XlaComputation CreateR0F32IdentityComputation() { in CreateR0F32IdentityComputation()
44 XlaComputation CreateR1S0F32AdditionComputation() { in CreateR1S0F32AdditionComputation()
54 XlaComputation CreateR1S2F32AdditionComputation() { in CreateR1S2F32AdditionComputation()
64 XlaComputation CreateR0F32TupleComputation() { in CreateR0F32TupleComputation()
79 XlaComputation callee = CreateR0F32IdentityComputation(); in XLA_TEST_F()
88 XlaComputation callee = CreateR1S0F32AdditionComputation(); in XLA_TEST_F()
98 XlaComputation callee = CreateR1S2F32AdditionComputation(); in XLA_TEST_F()
114 TF_ASSERT_OK_AND_ASSIGN(XlaComputation inner, builder.Build()); in XLA_TEST_F()
123 TF_ASSERT_OK_AND_ASSIGN(XlaComputation outer, builder2.Build()); in XLA_TEST_F()
141 XlaComputation callee = CreateR0F32TupleComputation(); in XLA_TEST_F()
|
| D | while_test.cc | 55 XlaComputation condition; in XLA_TEST_F()
64 XlaComputation body; in XLA_TEST_F()
91 XlaComputation condition; in XLA_TEST_F()
100 XlaComputation body; in XLA_TEST_F()
122 XlaComputation condition; in XLA_TEST_F()
131 XlaComputation body; in XLA_TEST_F()
154 XlaComputation condition; in XLA_TEST_F()
163 XlaComputation body; in XLA_TEST_F()
191 XlaComputation add; in XLA_TEST_F()
202 XlaComputation condition; in XLA_TEST_F()
[all …]
|
| D | compilation_cache_test.cc | 41 void ExecuteComputationR0F32(const XlaComputation& computation, in ExecuteComputationR0F32()
57 const XlaComputation& computation, in ExecuteComputationR2F32()
79 XlaComputation computation = builder.Build().ConsumeValueOrDie(); in XLA_TEST_F()
101 XlaComputation computation = builder.Build().ConsumeValueOrDie(); in XLA_TEST_F()
117 XlaComputation computation_neg = builder_neg.Build().ConsumeValueOrDie(); in XLA_TEST_F()
121 XlaComputation computation_exp = builder_exp.Build().ConsumeValueOrDie(); in XLA_TEST_F()
126 XlaComputation computation_add = builder_add.Build().ConsumeValueOrDie(); in XLA_TEST_F()
156 XlaComputation computation = builder.Build().ConsumeValueOrDie(); in XLA_TEST_F()
|
| D | conditional_test.cc | 28 XlaComputation CreateR0ConstantComputation(float value) { in CreateR0ConstantComputation()
37 XlaComputation CreateR0IdentityComputation() { in CreateR0IdentityComputation()
45 XlaComputation CreateCeilComputation(const Shape& shape) { in CreateCeilComputation()
54 XlaComputation CreateR0CeilComputation() { in CreateR0CeilComputation()
58 XlaComputation CreateR1CeilComputation() { in CreateR1CeilComputation()
62 XlaComputation CreateFloorComputation(const Shape& shape) { in CreateFloorComputation()
71 XlaComputation CreateR0FloorComputation() { in CreateR0FloorComputation()
75 XlaComputation CreateR1FloorComputation() { in CreateR1FloorComputation()
79 XlaComputation CreateTupleCeilComputation(const string& computation_name, in CreateTupleCeilComputation()
93 XlaComputation CreateR0TupleCeilComputation() { in CreateR0TupleCeilComputation()
[all …]
|
| D | replay_test.cc | 44 XlaComputation computation = builder.Build().ConsumeValueOrDie(); in TEST_F()
51 XlaComputation replayed = client_->LoadSnapshot(*module).ConsumeValueOrDie(); in TEST_F()
77 XlaComputation computation = builder.Build().ConsumeValueOrDie(); in XLA_TEST_F()
84 XlaComputation replayed = client_->LoadSnapshot(*module).ConsumeValueOrDie(); in XLA_TEST_F()
118 XlaComputation plus_two = plus_two_builder.Build().ConsumeValueOrDie(); in TEST_F()
124 XlaComputation computation = mapper_builder.Build().ConsumeValueOrDie(); in TEST_F()
131 XlaComputation replayed = client_->LoadSnapshot(*module).ConsumeValueOrDie(); in TEST_F()
|
| D | outfeed_in_nested_computation_test.cc | 41 TF_ASSERT_OK_AND_ASSIGN(XlaComputation loop_cond, [&] { in XLA_TEST_F()
51 TF_ASSERT_OK_AND_ASSIGN(XlaComputation loop_body, [&] { in XLA_TEST_F()
71 TF_ASSERT_OK_AND_ASSIGN(XlaComputation computation, b.Build()); in XLA_TEST_F()
127 TF_ASSERT_OK_AND_ASSIGN(XlaComputation true_computation, [&] { in XLA_TEST_F()
135 TF_ASSERT_OK_AND_ASSIGN(XlaComputation false_computation, [&] { in XLA_TEST_F()
146 TF_ASSERT_OK_AND_ASSIGN(XlaComputation computation, b.Build()); in XLA_TEST_F()
|
| D | local_client_test_base.h | 99 const XlaComputation& computation,
102 const XlaComputation& computation,
108 const XlaComputation& computation,
111 const XlaComputation& computation,
|
| D | map_test.cc | 53 XlaComputation CreateAdderToOne() { in CreateAdderToOne()
63 XlaComputation CreateMax() { in CreateMax()
76 XlaComputation CreateScalarOne() { in CreateScalarOne()
90 XlaComputation CreateMulByTwo() { in CreateMulByTwo()
108 XlaComputation CreateAdderToOneTimesItself() { in CreateAdderToOneTimesItself()
125 XlaComputation CreateMapPlusN(const XlaComputation& embedded_computation, in CreateMapPlusN()
139 XlaComputation CreateGt() { in CreateGt()
156 XlaComputation CreateTernaryAdder() { in CreateTernaryAdder()
450 XlaComputation max_with_square; in TEST_F()
491 StatusOr<XlaComputation> computation_status = builder.Build(); in TEST_F()
|
| D | reduce_test.cc | 68 using FuncGeneratorForType = XlaComputation (*)(PrimitiveType, XlaBuilder*);
70 using FuncGenerator = XlaComputation (*)(XlaBuilder*);
97 XlaComputation add_f32 = CreateScalarAddComputation(F32, &builder); in RunR1ToR0Test()
128 XlaComputation reduce; in RunR1ToR0PredTest()
166 XlaComputation reduce_op; in RunR2ToR1PredTest()
205 XlaComputation add_f32 = CreateScalarAddComputation(F32, &builder); in RunR2ToR0Test()
232 XlaComputation add_f32 = CreateScalarAddComputation(F32, &builder); in RunR2ToR1Test()
279 const std::function<XlaComputation(XlaBuilder*)>& in RunVectorizedReduceTestForType()
287 XlaComputation reduction_function = reduction_function_generator(&builder); in RunVectorizedReduceTestForType()
321 const std::function<XlaComputation(PrimitiveType, XlaBuilder*)>& in RunVectorizedReduceTest()
[all …]
|
| D | bad_rng_shape_validation_test.cc | 43 StatusOr<XlaComputation> computation = builder.Build(); in TEST_F()
60 StatusOr<XlaComputation> computation = builder.Build(); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_context.h | 87 const xla::XlaComputation* GetOrCreateMax(const DataType type);
92 const xla::XlaComputation* GetOrCreateMin(const DataType type);
97 const xla::XlaComputation* GetOrCreateAdd(const DataType type);
102 const xla::XlaComputation* GetOrCreateMul(const DataType type);
127 using ComputationMap = std::map<DataType, xla::XlaComputation>;
132 const xla::XlaComputation* LookupOrCreate(
134 const std::function<xla::XlaComputation()>& create);
|
| D | xla_context.cc | 85 const xla::XlaComputation* XlaContext::GetOrCreateMax(const DataType type) { in GetOrCreateMax()
101 const xla::XlaComputation* XlaContext::GetOrCreateMin(const DataType type) { in GetOrCreateMin()
117 const xla::XlaComputation* XlaContext::GetOrCreateAdd(const DataType type) { in GetOrCreateAdd()
133 const xla::XlaComputation* XlaContext::GetOrCreateMul(const DataType type) { in GetOrCreateMul()
149 const xla::XlaComputation* XlaContext::LookupOrCreate( in LookupOrCreate()
151 const std::function<xla::XlaComputation()>& create) { in LookupOrCreate()
|
| D | tf2xla.h | 36 xla::XlaComputation* computation);
46 xla::XlaComputation* computation, absl::string_view debug_info_filename,
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | unique_op.cc | 70 xla::XlaComputation BuildInnerLoopCond(XlaOpKernelContext* ctx, in BuildInnerLoopCond()
81 xla::XlaComputation BuildInnerLoopBody(XlaOpKernelContext* ctx, in BuildInnerLoopBody()
102 xla::XlaComputation BuildOuterLoopCond(XlaOpKernelContext* ctx, in BuildOuterLoopCond()
115 xla::XlaComputation BuildOuterLoopBody( in BuildOuterLoopBody()
117 xla::Shape single_element_shape, const xla::XlaComputation& inner_cond, in BuildOuterLoopBody()
118 const xla::XlaComputation& inner_body) { in BuildOuterLoopBody()
237 xla::XlaComputation inner_loop_cond = in Compile()
239 xla::XlaComputation inner_loop_body = in Compile()
241 xla::XlaComputation outer_loop_cond = in Compile()
243 xla::XlaComputation outer_loop_body = in Compile()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| D | mlir_hlo_builder.h | 145 const XlaComputation& computation,
150 const XlaComputation& computation,
167 const XlaComputation& comparator,
171 const XlaComputation& condition,
172 const XlaComputation& body,
186 const XlaComputation& update_computation,
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_compiler.cc | 75 const XlaComputation& computation) { in GetComputationSerializedProto()
84 const XlaComputation& computation) { in GetHloModule()
95 StatusOr<std::string> GetComputationHloText(const XlaComputation& computation) { in GetComputationHloText()
106 const XlaComputation& computation) { in GetComputationHloDotGraph()
115 StatusOr<uint64> HashComputation(const XlaComputation& computation) { in HashComputation()
277 py::class_<XlaComputation>(m, "XlaComputation") in BuildXlaCompilerSubmodule()
279 -> std::unique_ptr<XlaComputation> { in BuildXlaCompilerSubmodule()
282 return absl::make_unique<XlaComputation>(proto); in BuildXlaCompilerSubmodule()
285 .def("program_shape", &XlaComputation::GetProgramShape) in BuildXlaCompilerSubmodule()
|