Home
last modified time | relevance | path

Searched refs:output_dtype (Results 1 – 25 of 36) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/tests/
Dcategorical_op_test.py63 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)
144 for output_dtype in self.output_dtypes():
[all …]
/external/tensorflow/tensorflow/contrib/seq2seq/python/kernel_tests/
Dbasic_decoder_test.py69 output_dtype = my_decoder.output_dtype
76 output_dtype)
116 self.assertEqual(output_dtype.sample_id,
148 output_dtype = my_decoder.output_dtype
155 output_dtype)
193 self.assertEqual(output_dtype.sample_id,
224 output_dtype = my_decoder.output_dtype
231 output_dtype)
263 self.assertEqual(output_dtype.sample_id, sample_ids.dtype)
297 output_dtype = my_decoder.output_dtype
[all …]
Dbasic_decoder_v2_test.py78 output_dtype = my_decoder.output_dtype
85 output_dtype)
124 self.assertEqual(output_dtype.sample_id,
155 output_dtype = my_decoder.output_dtype
162 output_dtype)
199 self.assertEqual(output_dtype.sample_id,
231 output_dtype = my_decoder.output_dtype
238 output_dtype)
269 self.assertEqual(output_dtype.sample_id, sample_ids.dtype)
304 output_dtype = my_decoder.output_dtype
[all …]
/external/tensorflow/tensorflow/python/ops/
Dstateless_random_ops.py189 output_dtype=dtypes.int64, argument
221 output_dtype, seed)
267 logits, num_samples, seed, output_dtype=dtype)
Drandom_ops.py333 def multinomial(logits, num_samples, seed=None, name=None, output_dtype=None): argument
357 return multinomial_categorical_impl(logits, num_samples, output_dtype, seed)
393 logits, num_samples, seed=seed1, seed2=seed2, output_dtype=dtype)
Dnn_ops.py3833 output_dtype=dtypes.int64, argument
3890 Targmax=output_dtype,
3904 output_dtype=None, argument
3910 "output_dtype", output_dtype, "Targmax", Targmax)
Dimage_ops_test.py4007 def _convert(self, original, original_dtype, output_dtype, expected): argument
4009 y_np = np.array(expected, dtype=output_dtype.as_numpy_dtype())
4013 y = image_ops.convert_image_dtype(image, output_dtype)
4014 self.assertTrue(y.dtype == output_dtype)
4016 if output_dtype in [
4020 image, output_dtype, saturate=True)
4021 self.assertTrue(y_saturate.dtype == output_dtype)
Darray_ops.py2412 output_dtype = ops.convert_to_tensor(args[0]).dtype.base_dtype
2420 mult_fact = ones(shapes, output_dtype)
/external/tensorflow/tensorflow/contrib/seq2seq/python/ops/
Ddecoder.py75 def output_dtype(self): member in Decoder
192 def output_dtype(self): member in BaseDecoder
343 decoder.output_dtype,
375 decoder.output_dtype)
Dbasic_decoder.py106 def output_dtype(self): member in BasicDecoder
218 def output_dtype(self): member in BasicDecoderV2
Dbeam_search_decoder.py742 def output_dtype(self): member in BeamSearchDecoder
917 def output_dtype(self): member in BeamSearchDecoderV2
Dsampler.py752 logits_2d, n, seed=seed, output_dtype=sample_dtype)
Dhelper.py107 logits_2d, n, seed=seed, output_dtype=sample_dtype)
/external/tensorflow/tensorflow/python/kernel_tests/random/
Dstateless_random_ops_test.py122 for output_dtype in dtypes.int32, dtypes.int64:
128 output_dtype=output_dtype)
Dmultinomial_op_test.py60 for output_dtype in [np.int32, np.int64]:
66 logits, num_samples, output_dtype=output_dtype))
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.random.pbtxt37 …argspec: "args=[\'logits\', \'num_samples\', \'seed\', \'name\', \'output_dtype\'], varargs=None, …
61 …argspec: "args=[\'logits\', \'num_samples\', \'seed\', \'output_dtype\', \'name\'], varargs=None, …
/external/tensorflow/tensorflow/lite/kernels/
Dmul.cc217 #define TF_LITE_MUL(type, opname, output_dtype) \ in EvalQuantized() argument
225 GetTensorData<output_dtype>(output)) in EvalQuantized()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dimage_resize_ops.cc599 DataType output_dtype; in ResizeBilinearGradOp() local
600 OP_REQUIRES_OK(ctx, ctx->GetAttr("T", &output_dtype)); in ResizeBilinearGradOp()
601 OP_REQUIRES_OK(ctx, DataTypeToPrimitiveType(output_dtype, &output_type_)); in ResizeBilinearGradOp()
Dresampler_ops.cc602 DataType output_dtype; in ResamplerGradOp() local
603 OP_REQUIRES_OK(ctx, ctx->GetAttr("T", &output_dtype)); in ResamplerGradOp()
/external/tensorflow/tensorflow/python/eager/
Dpywrap_tfe_src.cc2186 TF_DataType output_dtype = input_dtype; local
2195 output_dtype = desired_dtype;
2198 if (output_dtype != TF_INT32) {
2322 TF_DataType output_dtype = TFE_TensorHandleDataType(handle.get()); local
2323 if (desired_dtype >= 0 && desired_dtype != output_dtype) {
2324 if (tensorflow::IsCompatible(desired_dtype, output_dtype)) {
2329 output_dtype = TFE_TensorHandleDataType(handle.get());
2345 dtype_setter(output_dtype);
/external/tensorflow/tensorflow/python/ops/distributions/
Dcategorical.py278 logits_2d, n, seed=seed, output_dtype=sample_dtype)
/external/tensorflow/tensorflow/lite/python/
Dop_hint.py1008 output_dtype = (
1011 output_dtypes.append(output_dtype)
/external/tensorflow/tensorflow/core/framework/
Dop_kernel.cc497 int input_index, int output_index, DataType output_dtype, in forward_input() argument
530 if (input_dtype(input_index) != output_dtype) { in forward_input()
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.nn.pbtxt225 …argspec: "args=[\'input\', \'ksize\', \'strides\', \'padding\', \'data_format\', \'output_dtype\',…
/external/tensorflow/tensorflow/contrib/learn/python/learn/learn_io/
Ddata_feeder.py418 def output_dtype(self): member in DataFeeder

12