Searched refs:xla_compile (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/python/compiler/xla/ |
| D | jit.py | 123 def xla_compile(node_def): function
126 xla_compile = attr_value_pb2.AttrValue(b=compile_ops)
130 xla_compile,
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | build_xla_ops_pass_test.cc | 191 auto xla_compile = NodeWith(Op("_XlaCompile"), Attr("must_compile", false)); in TEST_F() local
193 NodeWith(Op("Switch"), Inputs(Out(0, xla_compile), Out(1, xla_compile))); in TEST_F()
311 auto xla_compile = NodeWith(Op("_XlaCompile")); in TEST_F() local
313 NodeWith(Op("Switch"), Inputs(Out(0, xla_compile), Out(1, xla_compile))); in TEST_F()
|
| D | build_xla_ops_pass.cc | 495 ops::_XlaCompile xla_compile(root.WithOpName("xla_compile"), in ReplaceNodeWithXlaCompileAndXlaRun() local
505 xla_compile.operation.node()->AddAttr(kXlaHasReferenceVarsAttr, has_ref_attr); in ReplaceNodeWithXlaCompileAndXlaRun()
507 CopyIncomingControlEdges(g, /*from=*/n, /*to=*/xla_compile.key.node())); in ReplaceNodeWithXlaCompileAndXlaRun()
516 xla_compile.key, n->output_types()); in ReplaceNodeWithXlaCompileAndXlaRun()
535 xla_compile.key, xla_compile.compilation_successful); in ReplaceNodeWithXlaCompileAndXlaRun()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | control_flow_util.py | 60 xla_compile = op.get_attr("_XlaCompile")
61 if xla_compile: return True
|
| D | gradients_util.py | 326 xla_compile = func.definition.attr["_XlaCompile"].b
332 xla_compile = op.get_attr("_XlaCompile")
337 xla_compile = False
339 if not xla_compile:
352 "_XlaCompile": attr_value_pb2.AttrValue(b=xla_compile),
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | benchmarks_test.py | 915 xla_compile=False): argument
920 math_ops.reduce_logsumexp, jit_compile=xla_compile)
959 device=GPU, defunc=True, xla_compile=True)
963 device=GPU, execution_mode=context.ASYNC, defunc=True, xla_compile=True)
|