| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_op_kernel.cc | 32 XlaOpKernelContext::XlaOpKernelContext(OpKernelContext* context) in XlaOpKernelContext() function in tensorflow::XlaOpKernelContext
35 bool XlaOpKernelContext::ValidateInputsAreSameShape(OpKernel* op) { in ValidateInputsAreSameShape()
39 XlaContext* XlaOpKernelContext::xla_context() const { in xla_context()
43 xla::XlaBuilder* XlaOpKernelContext::builder() const { in builder()
47 XlaCompiler* XlaOpKernelContext::compiler() const { in compiler()
70 const XlaExpression& XlaOpKernelContext::InputExpression(int index) { in InputExpression()
74 const XlaExpression& XlaOpKernelContext::InputExpression( in InputExpression()
79 xla::XlaOp XlaOpKernelContext::Input(int index) { in Input()
83 xla::XlaOp XlaOpKernelContext::Input(absl::string_view name) { in Input()
87 TensorShape XlaOpKernelContext::InputShape(int index) { in InputShape()
[all …]
|
| D | xla_op_registry_test.cc | 30 void Compile(XlaOpKernelContext* ctx) override { in Compile()
39 void Compile(XlaOpKernelContext* ctx) override { in Compile()
90 void Compile(XlaOpKernelContext* ctx) override { in Compile()
|
| D | xla_op_kernel.h | 28 class XlaOpKernelContext; variable
41 virtual void Compile(XlaOpKernelContext* context) = 0;
59 class XlaOpKernelContext {
61 explicit XlaOpKernelContext(OpKernelContext* context);
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | relu_op.cc | 32 void Compile(XlaOpKernelContext* ctx) override { in Compile()
44 void Compile(XlaOpKernelContext* ctx) override { in Compile()
58 void Compile(XlaOpKernelContext* ctx) override { in Compile()
73 void Compile(XlaOpKernelContext* ctx) override { in Compile()
89 void Compile(XlaOpKernelContext* ctx) override { in Compile()
109 void Compile(XlaOpKernelContext* ctx) override { in Compile()
|
| D | shape_op.cc | 38 void Compile(XlaOpKernelContext* ctx) override { in Compile()
57 void Compile(XlaOpKernelContext* ctx) override { in Compile()
77 void Compile(XlaOpKernelContext* ctx) override { in Compile()
93 void Compile(XlaOpKernelContext* ctx) override { in Compile()
118 void Compile(XlaOpKernelContext* ctx) override { in Compile()
168 void Compile(XlaOpKernelContext* ctx) override { in Compile()
225 void Compile(XlaOpKernelContext* ctx) override { in Compile()
239 void Compile(XlaOpKernelContext* ctx) override { in Compile()
|
| D | sendrecv_ops.cc | 32 void Compile(XlaOpKernelContext* ctx) override;
44 void SendOp::Compile(XlaOpKernelContext* ctx) { in Compile()
56 void Compile(XlaOpKernelContext* ctx) override;
75 void RecvOp::Compile(XlaOpKernelContext* ctx) { in Compile()
|
| D | tensor_list_ops.cc | 49 void Compile(XlaOpKernelContext* ctx) override { in Compile()
64 Status CreateZerosList(XlaOpKernelContext* ctx, int element_shape_index, in CreateZerosList()
104 void Compile(XlaOpKernelContext* ctx) override { in Compile()
136 void Compile(XlaOpKernelContext* ctx) override { in Compile()
173 void Compile(XlaOpKernelContext* ctx) override { in Compile()
212 void Compile(XlaOpKernelContext* ctx) override { in Compile()
251 void Compile(XlaOpKernelContext* ctx) override { in Compile()
272 void Compile(XlaOpKernelContext* ctx) override { in Compile()
348 void Compile(XlaOpKernelContext* ctx) override { in Compile()
391 void Compile(XlaOpKernelContext* ctx) override { in Compile()
[all …]
|
| D | variable_ops.cc | 33 void Compile(XlaOpKernelContext* ctx) override { in Compile()
48 void Compile(XlaOpKernelContext* ctx) override { in Compile()
69 void Compile(XlaOpKernelContext* ctx) override { in Compile()
84 void Compile(XlaOpKernelContext* ctx) override { in Compile()
94 void Compile(XlaOpKernelContext* ctx) override { in Compile()
110 void Compile(XlaOpKernelContext* ctx) override { in Compile()
126 void Compile(XlaOpKernelContext* ctx) override { in Compile()
160 void Compile(XlaOpKernelContext* context) override { in Compile()
|
| D | elu_op.cc | 33 void Compile(XlaOpKernelContext* ctx) override { in Compile()
47 void Compile(XlaOpKernelContext* ctx) override { in Compile()
66 void Compile(XlaOpKernelContext* ctx) override { in Compile()
85 void Compile(XlaOpKernelContext* ctx) override { in Compile()
|
| D | tensor_array_ops.cc | 148 void Compile(XlaOpKernelContext* ctx) override { in Compile()
197 void Compile(XlaOpKernelContext* ctx) override { in Compile()
251 void Compile(XlaOpKernelContext* ctx) override { in Compile()
294 void Compile(XlaOpKernelContext* ctx) override { in Compile()
361 void Compile(XlaOpKernelContext* ctx) override { in Compile()
448 void Compile(XlaOpKernelContext* ctx) override { in Compile()
488 void Compile(XlaOpKernelContext* ctx) override { in Compile()
557 void Compile(XlaOpKernelContext* ctx) override { in Compile()
577 void Compile(XlaOpKernelContext* ctx) override { in Compile()
610 void Compile(XlaOpKernelContext* ctx) override { in Compile()
|
| D | categorical_op.cc | 40 void Compile(XlaOpKernelContext* ctx) override { in Compile()
114 XlaOpKernelContext* ctx) { in GetLogUniforms()
141 XlaOpKernelContext* ctx) override { in GetLogUniforms()
160 void Compile(XlaOpKernelContext* ctx) override { in Compile()
|
| D | listdiff_op.cc | 40 void Compile(XlaOpKernelContext* context) override { in Compile()
72 Status ListDiff(XlaOpKernelContext* context) { in ListDiff()
102 Status ListDiffWithIndexType(XlaOpKernelContext* context, DataType idx_type) { in ListDiffWithIndexType()
|
| D | random_ops.cc | 42 void Compile(XlaOpKernelContext* ctx) override { in Compile()
68 void Compile(XlaOpKernelContext* ctx) override { in Compile()
212 void Compile(XlaOpKernelContext* ctx) override { in Compile()
245 void Compile(XlaOpKernelContext* ctx) override { in Compile()
273 void Compile(XlaOpKernelContext* ctx) override { in Compile()
|
| D | training_ops.cc | 33 void Compile(XlaOpKernelContext* ctx) override { in Compile()
79 void Compile(XlaOpKernelContext* ctx) override { in Compile()
123 void Compile(XlaOpKernelContext* ctx) override { in Compile()
181 void Compile(XlaOpKernelContext* ctx) override { in Compile()
237 void Compile(XlaOpKernelContext* ctx) override { in Compile()
280 void Compile(XlaOpKernelContext* ctx) override { in Compile()
336 void Compile(XlaOpKernelContext* ctx) override { in Compile()
411 void Compile(XlaOpKernelContext* ctx) override { in Compile()
496 void Compile(XlaOpKernelContext* ctx) override { in Compile()
567 void Compile(XlaOpKernelContext* ctx) override { in Compile()
[all …]
|
| D | diag_op.cc | 83 void Compile(XlaOpKernelContext* ctx) override { in Compile()
125 void Compile(XlaOpKernelContext* ctx) override { in Compile()
167 void Compile(XlaOpKernelContext* ctx) override { in Compile()
196 void Compile(XlaOpKernelContext* ctx) override { in Compile()
|
| D | stack_ops.cc | 98 void Compile(XlaOpKernelContext* ctx) override { in Compile()
133 void Compile(XlaOpKernelContext* ctx) override { in Compile()
180 void Compile(XlaOpKernelContext* ctx) override { in Compile()
232 void Compile(XlaOpKernelContext* ctx) override { in Compile()
|
| D | sort_ops.cc | 27 void Compile(XlaOpKernelContext* context) override { in Compile()
39 void Compile(XlaOpKernelContext* context) override { in Compile()
|
| D | bcast_ops.cc | 38 void Compile(XlaOpKernelContext* ctx) override { in Compile()
86 void Compile(XlaOpKernelContext* ctx) override { in Compile()
112 void Output(XlaOpKernelContext* ctx, int idx, const BCast::Vec& v) { in Output()
|
| D | cwise_ops.h | 59 XlaOpKernelContext* ctx, const xla::XlaOp& lhs,
64 void Compile(XlaOpKernelContext* ctx) override;
|
| D | while_op.cc | 43 XlaOpKernelContext* ctx, std::vector<XlaCompiler::Argument>* args, in MakeXlaCompilerArgumentsFromInputs()
105 void GetLoopInvariants(XlaOpKernelContext* ctx, in GetLoopInvariants()
123 XlaOpKernelContext* ctx, const NameAttrList& body_name_attr, in ConvertLoopInvariantsToConst()
153 XlaOpKernelContext* ctx, in VerifyBodyInputAndOutputShapeMatch()
180 XlaOpKernelContext* ctx, const XlaCompiler::CompilationResult& cond) { in BuildWrappedCond()
191 XlaOpKernelContext* ctx, const XlaCompiler::CompilationResult& body, in BuildWrappedBody()
223 xla::XlaOp BuildWhile(XlaOpKernelContext* ctx, in BuildWhile()
271 void XlaWhileOp::Compile(XlaOpKernelContext* ctx) { in Compile()
|
| D | xla_self_adjoint_eig_op.cc | 31 void Compile(XlaOpKernelContext* ctx) override { in Compile()
47 void Compile(XlaOpKernelContext* ctx) override { in Compile()
|
| D | conv_ops.cc | 53 void Compile(XlaOpKernelContext* ctx) override { in Compile()
100 void Compile(XlaOpKernelContext* ctx) override { in Compile()
158 void Compile(XlaOpKernelContext* ctx) override { in Compile()
|
| D | stateless_random_ops.cc | 75 void Compile(XlaOpKernelContext* ctx) override { in Compile()
116 void Compile(XlaOpKernelContext* ctx) override { in Compile()
164 void Compile(XlaOpKernelContext* ctx) override { in Compile()
205 void Compile(XlaOpKernelContext* ctx) override { in Compile()
|
| D | softmax_op.cc | 39 void Compile(XlaOpKernelContext* ctx) override { in Compile()
92 XlaOpKernelContext* ctx, DataType type, xla::PrimitiveType xla_type, in CrossEntropyWithLogits()
147 void Compile(XlaOpKernelContext* ctx) override { in Compile()
180 void Compile(XlaOpKernelContext* ctx) override { in Compile()
|
| /external/tensorflow/tensorflow/compiler/aot/ |
| D | aot_only_var_handle_op.cc | 28 void Compile(XlaOpKernelContext* context) override;
39 void XlaAotOnlyVarHandleOp::Compile(XlaOpKernelContext* context) { in Compile()
|