| /external/tensorflow/tensorflow/compiler/tests/ |
| D | categorical_op_test.py | 78 def _testRngIsNotConstant(self, rng, dtype, output_dtype): argument
97 def rng(dtype, output_dtype): argument
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | stateless_random_ops.py | 499 output_dtype=dtypes.int32, argument
830 output_dtype=dtypes.int64, argument
|
| D | random_ops.py | 475 def multinomial(logits, num_samples, seed=None, name=None, output_dtype=None): argument
|
| D | nn_ops.py | 5130 output_dtype=dtypes.int64, argument
5205 output_dtype=None, argument
|
| D | image_ops_test.py | 4799 def _convert(self, original, original_dtype, output_dtype, expected): argument
|
| /external/tensorflow/tensorflow/cc/experimental/libtf/runtime/ |
| D | runtime.cc | 142 DataType output_dtype = output_tensor_spec_proto.dtype(); in ImportModule() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | image_resize_ops.cc | 515 DataType output_dtype = in GeneralCompile() local
727 DataType output_dtype; in ResizeBilinearGradOp() local
|
| D | resampler_ops.cc | 600 DataType output_dtype; in ResamplerGradOp() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | mul.cc | 258 #define TF_LITE_MUL(type, opname, output_dtype) \ in EvalQuantized() argument
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | io_ops.cc | 206 for (const auto& output_dtype : output_dtypes) { in WriteMetadataFile() local
479 for (const auto& output_dtype : output_dtypes) { in WriteMetadataFile() local
|
| D | snapshot_dataset_op.cc | 519 for (const auto& output_dtype : dataset()->output_dtypes()) { in WriteMetadataFile() local
1712 for (const auto& output_dtype : dataset()->output_dtypes()) { in GetNextInternal() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | convert_tensor.cc | 397 DataType output_dtype; in ConvertToTensorProto() local
|
| /external/tensorflow/tensorflow/core/ir/importexport/ |
| D | convert_tensor.cc | 396 tensorflow::DataType output_dtype; in ConvertToTensorProto() local
|
| /external/tensorflow/tensorflow/python/kernel_tests/random/ |
| D | stateless_random_ops_test.py | 157 def wrap(op, logits, logits_dtype, output_dtype, seed): argument
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_launch_util.cc | 326 DataType output_dtype, const TensorShape& output_shape, in GetOrCreateTensorForOutput()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | execute.cc | 1275 const DataType& output_dtype, in CreateUnshapedOutput()
|
| /external/tensorflow/tensorflow/python/kernel_tests/array_ops/ |
| D | array_ops_test.py | 1533 def check_output_dtype(output_dtype): argument
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel.cc | 512 int input_index, int output_index, DataType output_dtype, in forward_input()
|
| /external/tensorflow/tensorflow/core/transforms/constant_folding/ |
| D | pass.cc | 805 Type output_dtype = in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | constant_folding.cc | 848 const auto output_dtype = output_props[0].dtype(); in MaterializeConstantValuedNode() local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 4790 DataType output_dtype{DataType::DT_INT32}; in ConvertArgMinMax() local
|
| D | convert_nodes_test.cc | 8439 NodeDef GetArgMinMaxNodeDef(DataType input_dtype, DataType output_dtype) { in GetArgMinMaxNodeDef()
|