| /external/tensorflow/tensorflow/compiler/tests/ |
| D | categorical_op_test.py | 63 output_dtype=dtypes.int32)
81 def _testRngIsNotConstant(self, rng, dtype, output_dtype): argument
85 x = rng(dtype, output_dtype)
100 def rng(dtype, output_dtype): argument
102 output_dtype=output_dtype)
105 for output_dtype in self.output_dtypes():
106 self._testRngIsNotConstant(rng, dtype, output_dtype)
110 for output_dtype in self.output_dtypes():
115 output_dtype=output_dtype)
145 for output_dtype in self.output_dtypes():
[all …]
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | Multinomial.pbtxt | 109 type_attr: "output_dtype"
146 name: "output_dtype"
172 type_attr: "output_dtype"
209 name: "output_dtype"
|
| D | DebugNumericSummaryV2.pbtxt | 38 type_attr: "output_dtype"
41 name: "output_dtype"
|
| D | StatelessMultinomial.pbtxt | 17 type_attr: "output_dtype"
53 name: "output_dtype"
|
| /external/tensorflow/tensorflow/python/keras/layers/preprocessing/ |
| D | categorical.py | 101 output_dtype = dtypes.int64
104 shape=output_shape, dtype=output_dtype)
106 return tensor_spec.TensorSpec(shape=output_shape, dtype=output_dtype)
|
| D | categorical_encoding.py | 132 output_dtype = K.floatx() if self._output_mode == TFIDF else dtypes.int64
133 return tensor_spec.TensorSpec(shape=output_shape, dtype=output_dtype)
|
| D | text_vectorization.py | 400 output_dtype = K.floatx() if self._output_mode == TFIDF else dtypes.int64
401 return tensor_spec.TensorSpec(shape=output_shape, dtype=output_dtype)
|
| /external/tensorflow/tensorflow/python/debug/lib/ |
| D | debug_v2_ops_test.py | 285 output_dtype=dtypes.float64))
295 output_dtype=dtypes.float64))
306 output_dtype=dtypes.float64)), x._id
342 output_dtype=dtypes.float64)), x._id
369 output_dtype=dtypes.float64)), x._id
403 output_dtype=dtypes.float64)), x._id
440 output_dtype=dtypes.float64)), x._id
468 output_dtype=dtypes.float64)), x._id
494 output_dtype=dtypes.float64)), x._id
509 output_dtype=dtypes.float64)), x._id
[all …]
|
| D | dumping_callback.py | 383 output_dtype=dtypes.float64),
484 output_dtype=dtypes.float64))
495 output_dtype=dtypes.float64))
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | stateless_random_ops.py | 195 output_dtype=dtypes.int64, argument
227 output_dtype, seed)
273 logits, num_samples, seed, output_dtype=dtype)
|
| D | random_ops.py | 392 def multinomial(logits, num_samples, seed=None, name=None, output_dtype=None): argument
416 return multinomial_categorical_impl(logits, num_samples, output_dtype, seed)
452 logits, num_samples, seed=seed1, seed2=seed2, output_dtype=dtype)
|
| D | nn_ops.py | 4064 output_dtype=dtypes.int64, argument
4121 Targmax=output_dtype,
4135 output_dtype=None, argument
4141 "output_dtype", output_dtype, "Targmax", Targmax)
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | convert_tensor.cc | 296 DataType output_dtype; in ConvertToTensorProto() local
297 TF_RETURN_IF_ERROR(ConvertToDataType(type, &output_dtype)); in ConvertToTensorProto()
298 output_tensor->set_dtype(output_dtype); in ConvertToTensorProto()
301 switch (output_dtype) { in ConvertToTensorProto()
|
| /external/tensorflow/tensorflow/python/kernel_tests/random/ |
| D | stateless_random_ops_test.py | 122 for output_dtype in dtypes.int32, dtypes.int64:
128 output_dtype=output_dtype)
|
| D | multinomial_op_test.py | 60 for output_dtype in [np.int32, np.int64]:
66 logits, num_samples, output_dtype=output_dtype))
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.random.pbtxt | 53 …argspec: "args=[\'logits\', \'num_samples\', \'seed\', \'name\', \'output_dtype\'], varargs=None, …
81 …argspec: "args=[\'logits\', \'num_samples\', \'seed\', \'output_dtype\', \'name\'], varargs=None, …
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | mul.cc | 219 #define TF_LITE_MUL(type, opname, output_dtype) \ in EvalQuantized() argument
227 GetTensorData<output_dtype>(output)) in EvalQuantized()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | image_resize_ops.cc | 635 DataType output_dtype; in ResizeBilinearGradOp() local
636 OP_REQUIRES_OK(ctx, ctx->GetAttr("T", &output_dtype)); in ResizeBilinearGradOp()
637 OP_REQUIRES_OK(ctx, DataTypeToPrimitiveType(output_dtype, &output_type_)); in ResizeBilinearGradOp()
|
| D | resampler_ops.cc | 602 DataType output_dtype; in ResamplerGradOp() local
603 OP_REQUIRES_OK(ctx, ctx->GetAttr("T", &output_dtype)); in ResamplerGradOp()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_launch_util.cc | 311 DataType output_dtype, const TensorShape& output_shape, in GetOrCreateTensorForOutput() argument
318 return MakeTensor(output_dtype, output_shape, output_buffer, in GetOrCreateTensorForOutput()
|
| /external/tensorflow/tensorflow/python/ops/distributions/ |
| D | categorical.py | 278 logits_2d, n, seed=seed, output_dtype=sample_dtype)
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_DebugNumericSummaryV2.pbtxt | 76 name: "output_dtype"
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | op_hint.py | 1015 output_dtype = (
1018 output_dtypes.append(output_dtype)
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | pfor.py | 2800 output_dtype = pfor_input.get_attr("output_dtype")
2813 output_dtype=output_dtype)
2821 output_dtype=output_dtype)
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.nn.pbtxt | 225 …argspec: "args=[\'input\', \'ksize\', \'strides\', \'padding\', \'data_format\', \'output_dtype\',…
|