Searched defs:Tout (Results 1 – 9 of 9) sorted by relevance
/external/tensorflow/tensorflow/python/ops/ |
D | script_ops.py | 73 def __init__(self, func, Tout, is_grad_func): argument 275 Tout, argument 373 def eager_py_func(func, inp, Tout, name=None): argument 460 def py_func_common(func, inp, Tout, stateful=True, name=None): argument 554 def py_func(func, inp, Tout, stateful=True, name=None): argument 562 def numpy_function(func, inp, Tout, name=None): argument
|
/external/tensorflow/tensorflow/core/kernels/ |
D | cwise_ops_common.h | 88 typedef typename Functor::out_type Tout; // Output scalar data type. typedef 267 typedef typename Functor::out_type Tout; // Output scalar data type. typedef 298 typedef typename Functor::out_type Tout; // Output scalar data type. typedef
|
D | unique_op.cc | 125 auto Tout = output->flat<T>(); in Compute() local 176 auto Tout = output->shaped<T, 3>(new_sizes); in Compute() local
|
D | gather_nd_op_cpu_impl.h | 47 typename TTypes<T>::Matrix Tout, std::atomic<Index>* error_loc) in GatherNdSliceGenerator()
|
D | listdiff_op.cc | 72 auto Tout = out->vec<T>(); in Compute() local
|
D | argmax_op.cc | 164 #define DECLARE_GPU_SPEC(T, Tout, Dims) \ argument
|
D | dynamic_partition_op.cc | 44 OpOutputList* Tout) { in ValidateAndAllocateOutputs()
|
D | transpose_op.cc | 61 auto Tout = output->vec<T>(); in Compute() local
|
D | dynamic_partition_op_gpu.cu.cc | 224 OpOutputList* Tout, DoneCallback done) { in AllocateOutputs()
|