/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/core/ops/compat/ops_history_v2/ |
D | Multinomial.pbtxt | 109 type_attr: "output_dtype" 146 name: "output_dtype" 172 type_attr: "output_dtype" 209 name: "output_dtype"
|
D | DebugNumericSummaryV2.pbtxt | 9 type_attr: "output_dtype" 12 name: "output_dtype"
|
D | IsotonicRegression.pbtxt | 9 type_attr: "output_dtype" 36 name: "output_dtype"
|
D | StatelessMultinomial.pbtxt | 17 type_attr: "output_dtype" 53 name: "output_dtype"
|
/external/tensorflow/tensorflow/python/debug/lib/ |
D | debug_v2_ops_test.py | 349 output_dtype=dtypes.float64)) 359 output_dtype=dtypes.float64)) 370 output_dtype=dtypes.float64)), x._id 406 output_dtype=dtypes.float64)), x._id 433 output_dtype=dtypes.float64)), x._id 467 output_dtype=dtypes.float64)), x._id 504 output_dtype=dtypes.float64)), x._id 532 output_dtype=dtypes.float64)), x._id 558 output_dtype=dtypes.float64)), x._id 573 output_dtype=dtypes.float64)), x._id [all …]
|
D | dumping_callback.py | 431 output_dtype=dtypes.float64), **debug_identity_op_kwargs) 522 output_dtype=dtypes.float64)) 533 output_dtype=dtypes.float64))
|
/external/tensorflow/tensorflow/python/keras/layers/preprocessing/ |
D | hashing.py | 212 output_dtype = dtypes.int64 215 shape=output_shape, dtype=output_dtype) 217 return tensor_spec.TensorSpec(shape=output_shape, dtype=output_dtype)
|
D | category_encoding.py | 143 output_dtype = K.floatx() if self.output_mode == TFIDF else dtypes.int64 146 shape=output_shape, dtype=output_dtype) 148 return tensor_spec.TensorSpec(shape=output_shape, dtype=output_dtype)
|
D | discretization.py | 225 output_dtype = dtypes.int64 228 shape=output_shape, dtype=output_dtype) 229 return tensor_spec.TensorSpec(shape=output_shape, dtype=output_dtype)
|
D | text_vectorization.py | 377 output_dtype = dtypes.int64 if self._output_mode == INT else K.floatx() 378 return tensor_spec.TensorSpec(shape=output_shape, dtype=output_dtype)
|
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
D | convert_tensor.cc | 312 DataType output_dtype; in ConvertToTensorProto() local 313 TF_RETURN_IF_ERROR(ConvertToDataType(type, &output_dtype)); in ConvertToTensorProto() 314 output->set_dtype(output_dtype); in ConvertToTensorProto() 323 switch (output_dtype) { in ConvertToTensorProto() 370 DataTypeString(output_dtype))); in ConvertToTensorProto()
|
/external/tensorflow/tensorflow/python/ops/ |
D | stateless_random_ops.py | 256 output_dtype=dtypes.int32, argument 315 shape=shape, seed=seed, counts=counts, probs=probs, dtype=output_dtype) 587 output_dtype=dtypes.int64, argument 620 output_dtype, seed) 669 logits, num_samples, seed, output_dtype=dtype)
|
D | random_ops.py | 472 def multinomial(logits, num_samples, seed=None, name=None, output_dtype=None): argument 496 return multinomial_categorical_impl(logits, num_samples, output_dtype, seed) 533 logits, num_samples, seed=seed1, seed2=seed2, output_dtype=dtype)
|
/external/tensorflow/tensorflow/python/kernel_tests/random/ |
D | stateless_random_ops_test.py | 160 def wrap(op, logits, logits_dtype, output_dtype, seed): argument 163 num_samples=num_samples, output_dtype=output_dtype) 165 for output_dtype in dtypes.int32, dtypes.int64: 170 logits_dtype, output_dtype), 172 logits_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))
|
D | random_binomial_test.py | 87 shape=[1000], seed=[12, 34], counts=10., probs=0.4, output_dtype=dt) 89 shape=[1000], seed=[12, 34], counts=10., probs=0.4, output_dtype=dt)
|
/external/tensorflow/tensorflow/tools/api/golden/v1/ |
D | tensorflow.random.pbtxt | 53 …argspec: "args=[\'logits\', \'num_samples\', \'seed\', \'name\', \'output_dtype\'], varargs=None, … 77 …argspec: "args=[\'shape\', \'seed\', \'counts\', \'probs\', \'output_dtype\', \'name\'], varargs=N… 89 …argspec: "args=[\'logits\', \'num_samples\', \'seed\', \'output_dtype\', \'name\'], varargs=None, …
|
/external/tensorflow/tensorflow/core/api_def/base_api/ |
D | api_def_IsotonicRegression.pbtxt | 22 attr { name: "output_dtype" description: "Dtype of output." }
|
/external/tensorflow/tensorflow/lite/kernels/ |
D | mul.cc | 236 #define TF_LITE_MUL(type, opname, output_dtype) \ in EvalQuantized() argument 244 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()
|
/external/tensorflow/tensorflow/core/kernels/data/experimental/ |
D | io_ops.cc | 170 for (const auto& output_dtype : output_dtypes) { in WriteMetadataFile() local 171 metadata.add_dtype(output_dtype); in WriteMetadataFile()
|