| /external/tensorflow/tensorflow/compiler/tests/ |
| D | categorical_op_test.py | 81 def _testRngIsNotConstant(self, rng, dtype, output_dtype): argument
100 def rng(dtype, output_dtype): argument
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | stateless_random_ops.py | 256 output_dtype=dtypes.int32, argument
587 output_dtype=dtypes.int64, argument
|
| D | random_ops.py | 472 def multinomial(logits, num_samples, seed=None, name=None, output_dtype=None): argument
|
| D | nn_ops.py | 4776 output_dtype=dtypes.int64, argument
4848 output_dtype=None, argument
|
| D | image_ops_test.py | 4574 def _convert(self, original, original_dtype, output_dtype, expected): argument
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | mul.cc | 236 #define TF_LITE_MUL(type, opname, output_dtype) \ in EvalQuantized() argument
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | convert_tensor.cc | 312 DataType output_dtype; in ConvertToTensorProto() local
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | io_ops.cc | 170 for (const auto& output_dtype : output_dtypes) { in WriteMetadataFile() local
|
| D | snapshot_dataset_op.cc | 660 for (const auto& output_dtype : dataset()->output_dtypes()) { in WriteMetadataFile() local
1807 for (const auto& output_dtype : dataset()->output_dtypes()) { in GetNextInternal() local
|
| /external/tensorflow/tensorflow/python/kernel_tests/random/ |
| D | stateless_random_ops_test.py | 160 def wrap(op, logits, logits_dtype, output_dtype, seed): argument
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | image_resize_ops.cc | 635 DataType output_dtype; in ResizeBilinearGradOp() local
|
| D | resampler_ops.cc | 602 DataType output_dtype; in ResamplerGradOp() local
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_launch_util.cc | 306 DataType output_dtype, const TensorShape& output_shape, in GetOrCreateTensorForOutput()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | execute.cc | 576 const DataType& output_dtype, in CreateUnshapedOutput()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | array_ops_test.py | 1404 def check_output_dtype(output_dtype): argument
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel.cc | 499 int input_index, int output_index, DataType output_dtype, in forward_input()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | constant_folding.cc | 844 const auto output_dtype = output_props[0].dtype(); in MaterializeConstantValuedNode() local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 5678 DataType output_dtype = attrs.get<DataType>("output_type"); in ConvertArgMinMax() local
|
| D | convert_nodes_test.cc | 6372 NodeDef GetArgMinMaxNodeDef(DataType input_dtype, DataType output_dtype) { in TEST_P()
|