Home
last modified time | relevance | path

Searched refs:op_fn (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/python/kernel_tests/array_ops/
Dinplace_ops_test.py200 for op_fn in op_fns:
205 x = op_fn(x, [3], array_ops.ones([1, 0], dtype))
207 x = op_fn(x, None, array_ops.ones([1, 0], dtype))
/external/tensorflow/tensorflow/python/ops/
Dcond_v2.py265 op_fn = gen_functional_ops._if
267 op_fn = gen_functional_ops.stateless_if
270 if_op, tensors = util.get_op_and_outputs(op_fn(
1200 op_fn = gen_functional_ops.case
1202 op_fn = gen_functional_ops.stateless_case
1209 case_op, tensors = util.get_op_and_outputs(op_fn(
Dwhile_v2.py479 op_fn = gen_functional_ops._while
481 op_fn = gen_functional_ops.stateless_while
484 while_op, tensors = util.get_op_and_outputs(op_fn(
/external/tensorflow/tensorflow/python/eager/
Dforwardprop_test.py516 def testElementwiseNNOps(self, value, op_fn): argument
517 _test_gradients(self, op_fn, [constant_op.constant(value)], order=3)
/external/rust/crates/ppv-lite86/src/x86_64/
Dsse2.rs1539 ($vec:ident, $Op:ident, $op_fn:ident, $impl_fn:ident) => {
1543 fn $op_fn(self, rhs: Self) -> Self::Output {