Searched defs:out_dtype (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | lower_upper_bound_ops.cc | 33 void BuildLowerUpperBoundOp(XlaOpKernelContext* ctx, DataType out_dtype, in BuildLowerUpperBoundOp()
|
| /external/tensorflow/tensorflow/python/kernel_tests/random/ |
| D | stateless_random_ops_test.py | 190 def wrap(op, lam, lam_dtype, out_dtype, shape, seed): argument
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client_test.py | 1598 def _CreateConstantComputation(self, in_dtype, out_dtype): argument
1666 def testMapEachElementToConstant(self, in_dtype, out_dtype): argument
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | math_grad.py | 1771 def _SparseMatMul(t1, t2, out_dtype, transpose_a=False, transpose_b=False): argument
|
| /external/tensorflow/tensorflow/core/util/ |
| D | example_proto_fast_parsing.cc | 1709 DataType out_dtype; in FastParseSingleExample() local
|