| /external/tensorflow/tensorflow/compiler/tests/ |
| categorical_op_test.py | 62 output_dtype=dtypes.int32)
80 def _testRngIsNotConstant(self, rng, dtype, output_dtype):
84 x = rng(dtype, output_dtype)
99 def rng(dtype, output_dtype):
101 output_dtype=output_dtype)
104 for output_dtype in self.output_dtypes():
105 self._testRngIsNotConstant(rng, dtype, output_dtype)
109 for output_dtype in self.output_dtypes():
114 output_dtype=output_dtype
[all...] |
| /external/tensorflow/tensorflow/contrib/seq2seq/python/kernel_tests/ |
| basic_decoder_test.py | 72 output_dtype = my_decoder.output_dtype
79 output_dtype)
119 self.assertEqual(output_dtype.sample_id,
151 output_dtype = my_decoder.output_dtype
158 output_dtype)
196 self.assertEqual(output_dtype.sample_id,
227 output_dtype = my_decoder.output_dtype
[all...] |
| /external/tensorflow/tensorflow/contrib/seq2seq/python/ops/ |
| decoder.py | 70 def output_dtype(self): member in class:Decoder
204 decoder.output_dtype,
231 decoder.output_dtype)
|
| basic_decoder.py | 105 def output_dtype(self): member in class:BasicDecoder
|
| beam_search_decoder.py | 282 def output_dtype(self): member in class:BeamSearchDecoder
|
| /external/tensorflow/tensorflow/python/ops/ |
| random_ops.py | 326 def multinomial(logits, num_samples, seed=None, name=None, output_dtype=None):
346 output_dtype: integer type to use for the output. Defaults to int64.
355 logits, num_samples, seed=seed1, seed2=seed2, output_dtype=output_dtype)
|
| image_ops_test.py | [all...] |
| array_ops.py | [all...] |
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| image_resize_ops.cc | 366 DataType output_dtype; local
367 OP_REQUIRES_OK(ctx, ctx->GetAttr("T", &output_dtype));
368 OP_REQUIRES_OK(ctx, DataTypeToPrimitiveType(output_dtype, &output_type_));
|
| /external/tensorflow/tensorflow/python/kernel_tests/random/ |
| multinomial_op_test.py | 60 for output_dtype in [np.int32, np.int64]:
66 logits, num_samples, output_dtype=output_dtype))
|
| /external/tensorflow/tensorflow/python/ops/distributions/ |
| categorical.py | 270 logits_2d, n, seed=seed, output_dtype=sample_dtype)
|
| /external/tensorflow/tensorflow/core/framework/ |
| op_kernel.cc | 459 int input_index, DataType output_dtype, const TensorShape& output_shape,
469 if (input_dtype(input_index) != output_dtype) {
[all...] |
| /external/tensorflow/tensorflow/contrib/learn/python/learn/learn_io/ |
| data_feeder.py | 316 output_dtype: DType of output (or dictionary of shapes.
402 def output_dtype(self): member in class:DataFeeder
597 output_dtype: dtype of output (can be dictionary depending on `y`).
808 output_dtype: dtype of output.
|
