Searched refs:xla_ops (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/cc/ |
| D | BUILD | 11 out_ops_file = "ops/xla_ops",
12 deps = ["//tensorflow/compiler/tf2xla/ops:xla_ops"],
16 name = "xla_ops",
17 srcs = ["ops/xla_ops.cc"],
18 hdrs = ["ops/xla_ops.h"],
23 "//tensorflow/compiler/tf2xla/ops:xla_ops",
35 deps = ["//tensorflow/compiler/jit/ops:xla_ops"],
46 "//tensorflow/compiler/jit/ops:xla_ops",
|
| /external/tensorflow/tensorflow/compiler/jit/ops/ |
| D | BUILD | 10 name = "xla_ops",
11 srcs = ["xla_ops.cc"],
18 out = "xla_ops.py",
19 deps = ["//tensorflow/compiler/jit/ops:xla_ops"],
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ops/ |
| D | BUILD | 14 name = "xla_ops",
15 srcs = ["xla_ops.cc"],
30 ":xla_ops",
37 "xla_ops.cc",
|
| /external/tensorflow/tensorflow/compiler/jit/kernels/ |
| D | BUILD | 45 srcs = ["xla_ops.cc"],
46 hdrs = ["xla_ops.h"],
52 name = "xla_ops",
53 hdrs = ["xla_ops.h"],
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | BUILD | 83 "//tensorflow/compiler/jit/kernels:xla_ops",
85 "//tensorflow/compiler/tf2xla/kernels:xla_ops",
99 "//tensorflow/compiler/jit/kernels:xla_ops",
100 "//tensorflow/compiler/tf2xla/kernels:xla_ops",
127 "//tensorflow/compiler/jit/kernels:xla_ops",
130 "//tensorflow/compiler/tf2xla/kernels:xla_ops",
153 "//tensorflow/compiler/jit/kernels:xla_ops",
156 "//tensorflow/compiler/tf2xla/kernels:xla_ops",
178 "//tensorflow/compiler/jit/kernels:xla_ops",
231 "//tensorflow/compiler/jit/ops:xla_ops",
[all …]
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | xla_control_flow_ops_test.py | 22 from tensorflow.compiler.tf2xla.python import xla as xla_ops unknown
57 z1 = xla_ops.einsum(x_i, y, "ab,bc->ac")
58 z2 = xla_ops.einsum(x, y_i, "ab,bc->ac")
59 z3 = xla_ops.einsum(x, y, "ab,bc->ac")
60 z4 = xla_ops.einsum(x_i, y_i, "ab,bc->ac")
61 z5 = xla_ops.einsum(y_i, x_i, "cd,ce->de") # Includes transpose.
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | BUILD | 10 name = "xla_ops",
164 "//tensorflow/compiler/tf2xla/ops:xla_ops",
209 # of xla_ops. Therefore, resampler_ops is its own cc_library target, and its
280 "//tensorflow/compiler/tf2xla/ops:xla_ops",
298 "//tensorflow/compiler/tf2xla/ops:xla_ops",
320 "//tensorflow/compiler/tf2xla/ops:xla_ops",
340 "//tensorflow/compiler/tf2xla/ops:xla_ops",
|
| /external/tensorflow/tensorflow/core/tpu/kernels/xla/ |
| D | BUILD | 10 name = "xla_ops",
31 "//tensorflow/compiler/tf2xla/kernels:xla_ops",
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | BUILD | 57 "//tensorflow/compiler/tf2xla/kernels:xla_ops",
141 "//tensorflow/compiler/tf2xla/kernels:xla_ops",
750 "//tensorflow/compiler/tf2xla/kernels:xla_ops",
801 "//tensorflow/compiler/tf2xla/kernels:xla_ops",
819 "//tensorflow/compiler/tf2xla/ops:xla_ops",
841 "//tensorflow/compiler/tf2xla/ops:xla_ops",
869 "//tensorflow/compiler/tf2xla/ops:xla_ops",
930 "//tensorflow/compiler/tf2xla/ops:xla_ops",
967 "//tensorflow/compiler/tf2xla/ops:xla_ops",
993 "//tensorflow/compiler/tf2xla/cc:xla_ops",
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/python/ |
| D | BUILD | 30 "//tensorflow/compiler/tf2xla/ops:xla_ops",
|
| /external/tensorflow/tensorflow/python/compiler/xla/ |
| D | xla.py | 25 from tensorflow.compiler.jit.ops import xla_ops
393 xla_ops.xla_cluster_output(o, name='output{}'.format(i))
|
| /external/tensorflow/tensorflow/compiler/aot/ |
| D | BUILD | 56 "//tensorflow/compiler/tf2xla/kernels:xla_ops",
|