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 | 189 def wrap(op, lam, lam_dtype, out_dtype, shape, seed): argument
|
/external/tensorflow/tensorflow/compiler/xla/python/ |
D | xla_client_test.py | 1454 def _CreateConstantComputation(self, in_dtype, out_dtype): argument 1520 def testMapEachElementToConstant(self, in_dtype, out_dtype): argument
|
/external/tensorflow/tensorflow/python/ops/ |
D | math_grad.py | 1761 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
|