Home
last modified time | relevance | path

Searched refs:xla_ops (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/cc/
DBUILD11 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/
DBUILD10 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/jit/
DBUILD54 "//tensorflow/compiler/jit/kernels:xla_ops",
57 "//tensorflow/compiler/tf2xla/kernels:xla_ops",
68 "//tensorflow/compiler/jit/kernels:xla_ops",
69 "//tensorflow/compiler/tf2xla/kernels:xla_ops",
85 "//tensorflow/compiler/jit/kernels:xla_ops",
87 "//tensorflow/compiler/tf2xla/kernels:xla_ops",
104 "//tensorflow/compiler/jit/kernels:xla_ops",
106 "//tensorflow/compiler/tf2xla/kernels:xla_ops",
123 "//tensorflow/compiler/jit/kernels:xla_ops",
125 "//tensorflow/compiler/tf2xla/kernels:xla_ops",
[all …]
/external/tensorflow/tensorflow/compiler/tf2xla/ops/
DBUILD14 name = "xla_ops",
15 srcs = ["xla_ops.cc"],
29 ":xla_ops",
36 "xla_ops.cc",
/external/tensorflow/tensorflow/compiler/xla/python/
Dlocal_computation_builder.cc542 std::vector<XlaOp> xla_ops; in ConcatInDim() local
543 xla_ops.reserve(operands.size()); in ConcatInDim()
545 xla_ops.push_back(op.op()); in ConcatInDim()
547 return xla::ConcatInDim(&builder_, xla_ops, dimension); in ConcatInDim()
562 std::vector<XlaOp> xla_ops; in Tuple() local
563 xla_ops.reserve(elements.size()); in Tuple()
565 xla_ops.push_back(op.op()); in Tuple()
568 return xla::Tuple(&builder_, xla_ops); in Tuple()
610 std::vector<XlaOp> xla_ops; in Call() local
611 xla_ops.reserve(operands.size()); in Call()
[all …]
/external/tensorflow/tensorflow/compiler/jit/kernels/
DBUILD10 name = "xla_ops",
11 srcs = ["xla_ops.cc"],
12 hdrs = ["xla_ops.h"],
/external/tensorflow/tensorflow/compiler/tf2xla/
DBUILD37 "//tensorflow/compiler/tf2xla/kernels:xla_ops",
92 "//tensorflow/compiler/tf2xla/kernels:xla_ops",
395 "//tensorflow/compiler/tf2xla/kernels:xla_ops",
442 "//tensorflow/compiler/tf2xla/kernels:xla_ops",
459 "//tensorflow/compiler/tf2xla/ops:xla_ops",
481 "//tensorflow/compiler/tf2xla/ops:xla_ops",
508 "//tensorflow/compiler/tf2xla/ops:xla_ops",
544 "//tensorflow/compiler/tf2xla/ops:xla_ops",
568 "//tensorflow/compiler/tf2xla/cc:xla_ops",
593 "//tensorflow/compiler/tf2xla/cc:xla_ops",
[all …]
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
DBUILD10 name = "xla_ops",
134 "//tensorflow/compiler/tf2xla/ops:xla_ops",
202 # of xla_ops. Therefore, resampler_ops is its own cc_library target, and its
270 "//tensorflow/compiler/tf2xla/ops:xla_ops",
289 "//tensorflow/compiler/tf2xla/ops:xla_ops",
306 "//tensorflow/compiler/tf2xla/ops:xla_ops",
/external/tensorflow/tensorflow/compiler/tf2xla/python/
DBUILD36 "//tensorflow/compiler/tf2xla/ops:xla_ops",
/external/tensorflow/tensorflow/contrib/compiler/
Dxla.py25 from tensorflow.compiler.jit.ops import xla_ops
364 xla_ops.xla_cluster_output(o, name='output{}'.format(i))
/external/tensorflow/tensorflow/compiler/aot/
DBUILD42 "//tensorflow/compiler/tf2xla/kernels:xla_ops",
/external/tensorflow/tensorflow/contrib/rnn/python/kernel_tests/
Drnn_cell_test.py1959 xla_ops = _create_multi_lstm_cell_ops(
1967 xla_results = sess.run(xla_ops)