Searched refs:xla_compile (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/contrib/compiler/ |
| D | jit.py | 81 def xla_compile(node_def): function
84 xla_compile = attr_value_pb2.AttrValue(b=compile_ops)
88 xla_compile,
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | create_xla_launch_op.cc | 91 bool xla_compile = false; in CompilationRequested() local
93 node_def, kXlaCompileAttr, &xla_compile); in CompilationRequested()
94 if (!status.ok() || !xla_compile) { in CompilationRequested()
|
| D | build_xla_ops_pass.cc | 332 ops::_XlaCompile xla_compile(root.WithOpName("xla_compile"), in ReplaceNodeWithXlaCompileAndXlaRun() local
339 CopyIncomingControlEdges(g, /*from=*/n, /*to=*/xla_compile.key.node())); in ReplaceNodeWithXlaCompileAndXlaRun()
348 xla_compile.key, n->output_types()); in ReplaceNodeWithXlaCompileAndXlaRun()
367 xla_compile.key, xla_compile.compilation_successful); in ReplaceNodeWithXlaCompileAndXlaRun()
|
| D | build_xla_ops_pass_test.cc | 186 auto xla_compile = NodeWith(Op("_XlaCompile"), Attr("must_compile", false)); in TEST_F() local
188 NodeWith(Op("Switch"), Inputs(Out(0, xla_compile), Out(1, xla_compile))); in TEST_F()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | control_flow_util.py | 47 xla_compile = op.get_attr("_XlaCompile")
48 if xla_compile: return True
|
| D | gradients_util.py | 392 xla_compile = func.definition.attr["_XlaCompile"].b
398 xla_compile = op.get_attr("_XlaCompile")
405 if not xla_compile:
418 "_XlaCompile": attr_value_pb2.AttrValue(b=xla_compile),
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/tutorials/ |
| D | xla_compile.ipynb | 71 …" <a target=\"_blank\" href=\"https://www.tensorflow.org/xla/tutorials/xla_compile\"><img src=\…
74 …sorflow/tensorflow/blob/master/tensorflow/compiler/xla/g3doc/tutorials/xla_compile.ipynb\"><img sr…
77 …sorflow/tensorflow/blob/master/tensorflow/compiler/xla/g3doc/tutorials/xla_compile.ipynb\"><img sr…
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | jit.md | 89 [xla.compile() tutorial Colab](./tutorials/xla_compile.ipynb)
|