| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_op_kernel.cc | 33 XlaOpKernelContext::XlaOpKernelContext(OpKernelContext* context) in XlaOpKernelContext() function in tensorflow::XlaOpKernelContext
36 bool XlaOpKernelContext::ValidateInputsAreSameShape(OpKernel* op) { in ValidateInputsAreSameShape()
40 XlaContext* XlaOpKernelContext::xla_context() const { in xla_context()
44 xla::XlaBuilder* XlaOpKernelContext::builder() const { in builder()
48 XlaCompiler* XlaOpKernelContext::compiler() const { in compiler()
52 const XlaExpression& XlaOpKernelContext::InputExpression(int index) { in InputExpression()
56 const XlaExpression& XlaOpKernelContext::InputExpression( in InputExpression()
61 xla::XlaOp XlaOpKernelContext::Input(int index) { in Input()
65 xla::XlaOp XlaOpKernelContext::Input(absl::string_view name) { in Input()
69 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 | 31 class XlaOpKernelContext; variable
44 virtual void Compile(XlaOpKernelContext* context) = 0;
62 class XlaOpKernelContext {
64 explicit XlaOpKernelContext(OpKernelContext* context);
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | image_resize_ops.cc | 39 xla::Shape GetOutputShape(XlaOpKernelContext* ctx) const { in GetOutputShape()
54 void CompileGrad(XlaOpKernelContext* ctx, const char* target, in CompileGrad()
76 void CompileForward(XlaOpKernelContext* ctx, const char* target) { in CompileForward()
103 void Compile(XlaOpKernelContext* ctx) override { in Compile()
112 void Compile(XlaOpKernelContext* ctx) override { in Compile()
121 void Compile(XlaOpKernelContext* ctx) override { in Compile()
130 void Compile(XlaOpKernelContext* ctx) override { in Compile()
|
| D | cross_replica_ops.cc | 55 void Compile(XlaOpKernelContext* ctx) override { in Compile()
76 void Compile(XlaOpKernelContext* ctx) override { in Compile()
100 void Compile(XlaOpKernelContext* ctx) override { in Compile()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | relu_op.cc | 42 void Compile(XlaOpKernelContext* ctx) override { in Compile()
52 void Compile(XlaOpKernelContext* ctx) override { in Compile()
63 void Compile(XlaOpKernelContext* ctx) override { in Compile()
79 void Compile(XlaOpKernelContext* ctx) override { in Compile()
95 void Compile(XlaOpKernelContext* ctx) override { in Compile()
115 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 | scatter_nd_op.cc | 83 void Compile(XlaOpKernelContext* context) override { in Compile()
139 XlaOpKernelContext* context, in CompileTensorScatter()
177 void Compile(XlaOpKernelContext* context) override { in Compile()
190 void Compile(XlaOpKernelContext* context) override { in Compile()
203 void Compile(XlaOpKernelContext* context) override { in Compile()
216 void Compile(XlaOpKernelContext* context) override { in Compile()
229 void Compile(XlaOpKernelContext* context) override { in Compile()
|
| D | image_resize_ops.h | 26 void Compile(XlaOpKernelContext* ctx) override;
38 void Compile(XlaOpKernelContext* ctx) override;
50 void Compile(XlaOpKernelContext* ctx) override;
|
| D | tensor_list_ops.cc | 52 XlaOpKernelContext* ctx, const xla::Shape& element_shape, in GetTensorListDynamicDims()
91 void Compile(XlaOpKernelContext* ctx) override { in Compile()
110 Status TryGetElementShapeFromInput(XlaOpKernelContext* ctx, xla::XlaOp input, in TryGetElementShapeFromInput()
145 void Compile(XlaOpKernelContext* ctx) override { in Compile()
215 void Compile(XlaOpKernelContext* ctx) override { in Compile()
289 void Compile(XlaOpKernelContext* ctx) override { in Compile()
345 void Compile(XlaOpKernelContext* ctx) override { in Compile()
383 void Compile(XlaOpKernelContext* ctx) override { in Compile()
434 void Compile(XlaOpKernelContext* ctx) override { in Compile()
464 void Compile(XlaOpKernelContext* ctx) override { in Compile()
[all …]
|
| D | variable_ops.cc | 34 void Compile(XlaOpKernelContext* ctx) override { in Compile()
49 void Compile(XlaOpKernelContext* ctx) override { in Compile()
70 void Compile(XlaOpKernelContext* ctx) override { in Compile()
85 void Compile(XlaOpKernelContext* ctx) override { in Compile()
95 void Compile(XlaOpKernelContext* ctx) override { in Compile()
111 void Compile(XlaOpKernelContext* ctx) override { in Compile()
129 void Compile(XlaOpKernelContext* ctx) override { in Compile()
158 void Compile(XlaOpKernelContext* context) override { in Compile()
|
| D | shape_op.cc | 44 void Compile(XlaOpKernelContext* ctx) override { in Compile()
76 void Compile(XlaOpKernelContext* ctx) override { in Compile()
118 void Compile(XlaOpKernelContext* ctx) override { in Compile()
151 void Compile(XlaOpKernelContext* ctx) override { in Compile()
189 void Compile(XlaOpKernelContext* ctx) override { in Compile()
205 void Compile(XlaOpKernelContext* ctx) override { in Compile()
232 void Compile(XlaOpKernelContext* ctx) override { in Compile()
282 void Compile(XlaOpKernelContext* ctx) override { in Compile()
339 void Compile(XlaOpKernelContext* ctx) override { in Compile()
404 void Compile(XlaOpKernelContext* ctx) override { in Compile()
|
| D | elu_op.cc | 52 void Compile(XlaOpKernelContext* ctx) override { in Compile()
62 void Compile(XlaOpKernelContext* ctx) override { in Compile()
81 void Compile(XlaOpKernelContext* ctx) override { in Compile()
91 void Compile(XlaOpKernelContext* ctx) override { in Compile()
|
| D | lower_upper_bound_ops.cc | 33 void BuildLowerUpperBoundOp(XlaOpKernelContext* ctx, DataType out_dtype, in BuildLowerUpperBoundOp()
89 void Compile(XlaOpKernelContext* ctx) override { in Compile()
105 void Compile(XlaOpKernelContext* ctx) override { in Compile()
|
| D | replica_id_op.cc | 26 void Compile(XlaOpKernelContext* ctx) override;
32 void XlaReplicaIdOp::Compile(XlaOpKernelContext* ctx) { in Compile()
|
| D | if_while_utils.h | 33 XlaOpKernelContext* ctx, std::vector<XlaCompiler::Argument>* args,
39 Status FindMustBeConstNodes(XlaOpKernelContext* ctx,
|
| 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 | case_op.h | 48 void Compile(XlaOpKernelContext* ctx) override;
60 XlaOpKernelContext* ctx);
|
| D | categorical_op.cc | 43 void Compile(XlaOpKernelContext* ctx) override { in Compile()
131 XlaOpKernelContext* ctx) { in GetLogUniforms()
162 XlaOpKernelContext* ctx) override { in GetLogUniforms()
179 void Compile(XlaOpKernelContext* ctx) override { in Compile()
|
| D | random_ops.cc | 44 void Compile(XlaOpKernelContext* ctx) override { in Compile()
74 void Compile(XlaOpKernelContext* ctx) override { in Compile()
223 void Compile(XlaOpKernelContext* ctx) override { in Compile()
260 void Compile(XlaOpKernelContext* ctx) override { in Compile()
288 void Compile(XlaOpKernelContext* ctx) override { in Compile()
321 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 | 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()
239 void Compile(XlaOpKernelContext* ctx) override { in Compile()
288 void Compile(XlaOpKernelContext* ctx) override { in Compile()
343 void Compile(XlaOpKernelContext* ctx) override { in Compile()
399 void Compile(XlaOpKernelContext* ctx) override { in Compile()
475 void Compile(XlaOpKernelContext* ctx) override { in Compile()
571 void Compile(XlaOpKernelContext* ctx) override { in Compile()
[all …]
|
| 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()
116 XlaOpKernelContext* ctx, xla::Shape outer_loop_shape, in BuildOuterLoopBody()
146 xla::XlaOp DataOutputFastPath(XlaOpKernelContext* ctx, xla::XlaOp input, in DataOutputFastPath()
215 void Compile(XlaOpKernelContext* ctx) override { in Compile()
|
| D | cwise_ops.h | 59 XlaOpKernelContext* ctx, const xla::XlaOp& lhs,
64 void Compile(XlaOpKernelContext* ctx) override;
|
| 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()
|