Home
last modified time | relevance | path

Searched defs:Tout (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/python/ops/
Dscript_ops.py76 def __init__(self, func, Tout, is_grad_func): argument
303 Tout, argument
425 def eager_py_func(func, inp, Tout, name=None): argument
527 def py_func_common(func, inp, Tout, stateful=True, name=None): argument
680 def py_func(func, inp, Tout, stateful=True, name=None): argument
689 def numpy_function(func, inp, Tout, stateful=True, name=None): argument
778 def _wrap_for_composites(func, inp, Tout): argument
/external/tensorflow/tensorflow/core/kernels/
Dcwise_ops_common.h82 typedef typename Functor::out_type Tout; // Output scalar data type. typedef
271 typedef typename Functor::out_type Tout; // Output scalar data type. typedef
307 typedef typename Functor::out_type Tout; // Output scalar data type. typedef
Dunique_op.cc169 auto Tout = output->flat<T>(); in Compute() local
221 auto Tout = output->shaped<T, 3>(new_sizes); in Compute() local
Dlistdiff_op.cc72 auto Tout = out->vec<T>(); in Compute() local
Dgather_nd_op_cpu_impl.h47 typename TTypes<T>::Matrix Tout, std::atomic<Index>* error_loc) in GatherNdSliceGenerator()
Dcwise_op_leakyrelu_bf16.cc79 typedef typename Functor::out_type Tout; // Output scalar data type. typedef in tensorflow::LeakyReluOp
Dargmax_op.cc177 #define DECLARE_GPU_SPEC(T, Tout, Dims) \ argument
Ddynamic_partition_op.cc44 OpOutputList* Tout) { in ValidateAndAllocateOutputs()
Dtranspose_op.cc61 auto Tout = output->vec<T>(); in Compute() local
Disotonic_regression_op.cc208 #define REGISTER_CPU_KERNEL(Tin, Tout) \ argument
Ddynamic_partition_op_gpu.cu.cc209 OpOutputList* Tout, DoneCallback done) { in AllocateOutputs()
/external/tensorflow/tensorflow/compiler/tf2xla/python/
Dxla.py566 def call_module(args, *, module, Tout, Sout, dim_args_spec=()): argument