| /external/pytorch/torch/_higher_order_ops/ |
| D | out_dtype.py | 51 def __call__(self, op, output_dtype, *args): argument
80 def trace_out_dtype(proxy_mode, func_overload, op, output_dtype, *args): argument
108 def is_int_mm(op, output_dtype, args): argument
120 def out_dtype_fallback(op, output_dtype, *args): argument
161 def out_dtype_func(ctx, op, output_dtype, *args): argument
|
| /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/pytorch/torch/_inductor/codegen/ |
| D | cpp_micro_gemm.py | 72 output_dtype, argument
208 output_dtype=None, argument
253 self, name, input_dtype, input2_dtype, output_dtype, compute_dtype, alpha argument
751 output_dtype=None, argument
|
| /external/pytorch/torch/distributed/_composable/fsdp/ |
| D | _fsdp_api.py | 46 output_dtype: Optional[torch.dtype] = None variable in MixedPrecisionPolicy
|
| /external/pytorch/torch/_inductor/ |
| D | mkldnn_lowerings.py | 379 output_dtype, argument
426 output_dtype, argument
482 output_dtype, argument
739 output_dtype, argument
|
| D | mkldnn_ir.py | 744 output_dtype, argument
1019 output_dtype, argument
1444 output_dtype, argument
1700 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
|
| /external/pytorch/aten/src/ATen/native/ |
| D | BucketizationUtils.h | 158 ScalarType output_dtype = output.scalar_type(); in searchsorted_pre_check() local
|
| /external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| D | qlinear.cpp | 923 std::optional<c10::ScalarType> output_dtype, in linear_int8_with_onednn_weight()
1204 std::optional<c10::ScalarType> output_dtype, in run_pointwise()
1233 std::optional<c10::ScalarType> output_dtype, in run_pointwise_tensor()
1265 std::optional<c10::ScalarType> output_dtype, in run_pointwise_binary()
1297 std::optional<c10::ScalarType> output_dtype, in run_pointwise_binary_tensor()
|
| D | qconv.cpp | 1405 std::optional<c10::ScalarType> output_dtype, in _quantized_convolution_onednn()
1883 std::optional<c10::ScalarType> output_dtype, in run_pointwise()
1935 std::optional<c10::ScalarType> output_dtype, in run_pointwise_binary()
|
| /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/pytorch/torch/csrc/distributed/c10d/ |
| D | NCCLUtils.cpp | 608 for (const auto& output_dtype : e.output_dtypes_) { in getCollectiveTrace() local
746 for (const auto& output_dtype : e.output_dtypes_) { in dump_json() 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/pytorch/aten/src/ATen/native/cuda/ |
| D | ForeachReduceOp.cu | 446 const ScalarType output_dtype = in foreach_tensor_norm_cuda() local
|
| /external/pytorch/torch/ao/quantization/backend_config/ |
| D | backend_config.py | 224 def output_dtype(self) -> Optional[torch.dtype]: member in DTypeConfig
|
| /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()
|