Home
last modified time | relevance | path

Searched defs:output_dtype (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/compiler/tests/
Dcategorical_op_test.py81 def _testRngIsNotConstant(self, rng, dtype, output_dtype): argument
100 def rng(dtype, output_dtype): argument
/external/tensorflow/tensorflow/contrib/seq2seq/python/ops/
Dbasic_decoder.py106 def output_dtype(self): member in BasicDecoder
218 def output_dtype(self): member in BasicDecoderV2
Ddecoder.py75 def output_dtype(self): member in Decoder
192 def output_dtype(self): member in BaseDecoder
Dbeam_search_decoder.py742 def output_dtype(self): member in BeamSearchDecoder
917 def output_dtype(self): member in BeamSearchDecoderV2
/external/tensorflow/tensorflow/python/ops/
Dstateless_random_ops.py189 output_dtype=dtypes.int64, argument
Drandom_ops.py333 def multinomial(logits, num_samples, seed=None, name=None, output_dtype=None): argument
Dnn_ops.py3833 output_dtype=dtypes.int64, argument
3904 output_dtype=None, argument
Dimage_ops_test.py4007 def _convert(self, original, original_dtype, output_dtype, expected): argument
/external/tensorflow/tensorflow/lite/kernels/
Dmul.cc217 #define TF_LITE_MUL(type, opname, output_dtype) \ in EvalQuantized() argument
/external/tensorflow/tensorflow/contrib/learn/python/learn/learn_io/
Ddata_feeder.py418 def output_dtype(self): member in DataFeeder
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dimage_resize_ops.cc599 DataType output_dtype; in ResizeBilinearGradOp() local
Dresampler_ops.cc602 DataType output_dtype; in ResamplerGradOp() local
/external/tensorflow/tensorflow/python/eager/
Dpywrap_tfe_src.cc2186 TF_DataType output_dtype = input_dtype; local
2322 TF_DataType output_dtype = TFE_TensorHandleDataType(handle.get()); local
/external/tensorflow/tensorflow/core/framework/
Dop_kernel.cc497 int input_index, int output_index, DataType output_dtype, in forward_input()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dconstant_folding.cc786 const auto output_dtype = output_props[0].dtype(); in MaterializeConstantValuedNode() local