Home
last modified time | relevance | path

Searched refs:out_dtype (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/python/ops/
Dlinalg_ops.py406 out_dtype = dtypes.complex64
408 out_dtype = dtypes.complex128
409 e, v = gen_linalg_ops.eig(tensor, Tout=out_dtype, compute_v=True, name=name)
433 out_dtype = dtypes.complex64
435 out_dtype = dtypes.complex128
436 e, _ = gen_linalg_ops.eig(tensor, Tout=out_dtype, compute_v=False, name=name)
Dmath_grad.py1761 def _SparseMatMul(t1, t2, out_dtype, transpose_a=False, transpose_b=False): argument
1777 if prod.dtype != out_dtype:
1778 prod = math_ops.cast(prod, out_dtype)
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dlower_upper_bound_ops.cc33 void BuildLowerUpperBoundOp(XlaOpKernelContext* ctx, DataType out_dtype, in BuildLowerUpperBoundOp() argument
67 const DataType accumulation_type = XlaHelpers::SumAccumulationType(out_dtype); in BuildLowerUpperBoundOp()
/external/tensorflow/tensorflow/python/kernel_tests/random/
Dstateless_random_ops_test.py189 def wrap(op, lam, lam_dtype, out_dtype, shape, seed): argument
192 dtype=out_dtype)
194 for out_dtype in np.float16, np.float32, np.float64, np.int32, np.int64:
198 lam_dtype, out_dtype,
201 lam_dtype, out_dtype, (10,)))
/external/tensorflow/tensorflow/compiler/xla/python/
Dxla_client_test.py1454 def _CreateConstantComputation(self, in_dtype, out_dtype): argument
1457 in_dtype.__name__, out_dtype.__name__))
1460 ops.Constant(c, out_dtype(1))
1516 "testcase_name": "_{}_{}".format(in_dtype.__name__, out_dtype.__name__),
1518 "out_dtype": out_dtype,
1519 } for in_dtype, out_dtype in [[np.float32, np.int32]])
1520 def testMapEachElementToConstant(self, in_dtype, out_dtype): argument
1524 self._CreateConstantComputation(in_dtype, out_dtype), [0])
/external/tensorflow/tensorflow/python/ops/numpy_ops/
Dnp_math_ops.py1250 out_dtype = np_utils.result_type(a.dtype, weights)
1252 out_dtype = np_utils.result_type(a.dtype, weights,
1254 a = np_array_ops.array(a, out_dtype)
1255 weights = np_array_ops.array(weights, out_dtype)
/external/tensorflow/tensorflow/core/util/
Dexample_proto_fast_parsing.cc1709 DataType out_dtype; in FastParseSingleExample() local
1718 out_dtype = config.dense[d].dtype; in FastParseSingleExample()
1737 out_dtype = config.sparse[d].dtype; in FastParseSingleExample()
1741 out_dtype = config.ragged[d].dtype; in FastParseSingleExample()
1747 *out = Tensor(out_dtype, out_shape); in FastParseSingleExample()
1759 *out = Tensor(out_dtype, out_shape); in FastParseSingleExample()
/external/tensorflow/tensorflow/python/ops/parallel_for/
Dpfor.py637 out_dtype = body_output.dtype
646 lambda: constant_op.constant([], dtype=out_dtype))