Home
last modified time | relevance | path

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

123

/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)
145 for output_dtype in self.output_dtypes():
[all …]
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DMultinomial.pbtxt109 type_attr: "output_dtype"
146 name: "output_dtype"
172 type_attr: "output_dtype"
209 name: "output_dtype"
DDebugNumericSummaryV2.pbtxt38 type_attr: "output_dtype"
41 name: "output_dtype"
DStatelessMultinomial.pbtxt17 type_attr: "output_dtype"
53 name: "output_dtype"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DMultinomial.pbtxt109 type_attr: "output_dtype"
146 name: "output_dtype"
172 type_attr: "output_dtype"
209 name: "output_dtype"
DDebugNumericSummaryV2.pbtxt9 type_attr: "output_dtype"
12 name: "output_dtype"
DIsotonicRegression.pbtxt9 type_attr: "output_dtype"
36 name: "output_dtype"
DStatelessMultinomial.pbtxt17 type_attr: "output_dtype"
53 name: "output_dtype"
/external/tensorflow/tensorflow/python/debug/lib/
Ddebug_v2_ops_test.py349 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 …]
Ddumping_callback.py431 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/
Dhashing.py212 output_dtype = dtypes.int64
215 shape=output_shape, dtype=output_dtype)
217 return tensor_spec.TensorSpec(shape=output_shape, dtype=output_dtype)
Dcategory_encoding.py143 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)
Ddiscretization.py225 output_dtype = dtypes.int64
228 shape=output_shape, dtype=output_dtype)
229 return tensor_spec.TensorSpec(shape=output_shape, dtype=output_dtype)
Dtext_vectorization.py377 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/
Dconvert_tensor.cc312 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/
Dstateless_random_ops.py256 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)
Drandom_ops.py472 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/
Dstateless_random_ops_test.py160 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))
Dmultinomial_op_test.py60 for output_dtype in [np.int32, np.int64]:
66 logits, num_samples, output_dtype=output_dtype))
Drandom_binomial_test.py87 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/
Dtensorflow.random.pbtxt53 …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/
Dapi_def_IsotonicRegression.pbtxt22 attr { name: "output_dtype" description: "Dtype of output." }
/external/tensorflow/tensorflow/lite/kernels/
Dmul.cc236 #define TF_LITE_MUL(type, opname, output_dtype) \ in EvalQuantized() argument
244 GetTensorData<output_dtype>(output)) in EvalQuantized()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dimage_resize_ops.cc635 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/
Dio_ops.cc170 for (const auto& output_dtype : output_dtypes) { in WriteMetadataFile() local
171 metadata.add_dtype(output_dtype); in WriteMetadataFile()

123