Home
last modified time | relevance | path

Searched defs:out_dtype (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dlower_upper_bound_ops.cc33 void BuildLowerUpperBoundOp(XlaOpKernelContext* ctx, DataType 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
/external/tensorflow/tensorflow/compiler/xla/python/
Dxla_client_test.py1454 def _CreateConstantComputation(self, in_dtype, out_dtype): argument
1520 def testMapEachElementToConstant(self, in_dtype, out_dtype): argument
/external/tensorflow/tensorflow/python/ops/
Dmath_grad.py1761 def _SparseMatMul(t1, t2, out_dtype, transpose_a=False, transpose_b=False): argument
/external/tensorflow/tensorflow/core/util/
Dexample_proto_fast_parsing.cc1709 DataType out_dtype; in FastParseSingleExample() local