| /external/tensorflow/tensorflow/core/kernels/ |
| H A D | listdiff_op.cc | 72 auto Tout = out->vec<T>(); variable
86 Tout(p) = Tx(i);
|
| H A D | dynamic_partition_op.cc | 44 OpOutputList* Tout) {
69 OP_REQUIRES_OK(c, c->output_list("outputs", Tout));
77 OP_REQUIRES_OK(c, Tout->allocate(p, shape, &out));
42 ValidateAndAllocateOutputs(OpKernelContext* c, const Tensor** data, const Tensor** partitions, OpOutputList* Tout) argument
|
| H A D | transpose_op.cc | 61 auto Tout = output->vec<T>(); variable
62 std::fill_n(Tout.data(), N, -1);
67 OP_REQUIRES(context, Tout(d) == -1,
69 Tout(d) = i;
|
| H A D | gather_nd_op_cpu_impl.h | 47 typename TTypes<T>::Matrix Tout, std::atomic<Index>* error_loc)
51 Tout_(Tout),
102 typename TTypes<T>::Matrix Tout) {
115 slice_size, Tindices, Tparams, Tout, &error_loc);
133 typename TTypes<T>::Matrix Tout);
44 GatherNdSliceGenerator( const Index slice_size, typename TTypes<Index>::ConstMatrix Tindices, typename TTypes<T, IXDIM + 1>::ConstTensor Tparams, typename TTypes<T>::Matrix Tout, std::atomic<Index>* error_loc) argument
98 operator ()(const CPUDevice& d, const Index slice_size, typename TTypes<int32>::Scalar Tscratch, typename TTypes<T, IXDIM + 1>::ConstTensor Tparams, typename TTypes<Index>::ConstMatrix Tindices, typename TTypes<T>::Matrix Tout) argument
|
| H A D | gather_nd_op_gpu.cu.cc | 73 typename TTypes<T>::Matrix Tout) {
75 const int64 out_size = Tout.size();
76 int64 s_size = Tout.dimension(1);
92 Tparams.data(), Tindices.data(), Tout.data(), batch_strides,
69 operator ()(const GPUDevice& d, const Index unused_slice_size, typename TTypes<int32>::Scalar Tscratch, typename TTypes<T, IXDIM + 1>::ConstTensor Tparams, typename TTypes<Index>::ConstMatrix Tindices, typename TTypes<T>::Matrix Tout) argument
|
| H A D | cwise_ops_gpu_common.cu.h | 61 typedef typename Functor::out_type Tout; typedef
64 typedef typename Eigen::internal::scalar_left<Tout, Tin, Binary> Unary;
71 typedef typename Functor::out_type Tout; typedef
74 typedef typename Eigen::internal::scalar_right<Tout, Tin, Binary> Unary;
|
| H A D | cwise_ops_common.h | 83 typedef typename Functor::out_type Tout; // Output scalar data type. typedef in class:tensorflow::BinaryOp
86 : BinaryOpShared(ctx, DataTypeToEnum<Tout>::v(),
105 auto out_flat = out->flat<Tout>();
123 eigen_device, out->shaped<Tout, 2>(bcast->result_shape()),
130 eigen_device, out->shaped<Tout, 3>(bcast->result_shape()),
137 eigen_device, out->shaped<Tout, 4>(bcast->result_shape()),
144 eigen_device, out->shaped<Tout, 5>(bcast->result_shape()),
198 typedef typename Functor::out_type Tout; // Output scalar data type. typedef in class:tensorflow::SimpleBinaryOp
210 if (std::is_same<Tin, Tout>::value) {
216 auto out_flat = out->flat<Tout>();
229 typedef typename Functor::out_type Tout; // Output scalar data type. typedef in class:tensorflow::UnaryOp
291 typedef typename Functor::out_type Tout; typedef
301 typedef typename Functor::out_type Tout; typedef
347 typedef typename Functor::out_type Tout; typedef
357 typedef typename Functor::out_type Tout; typedef
487 typedef typename Functor::out_type Tout; typedef
497 typedef typename Functor::out_type Tout; typedef
[all...] |
| H A D | dynamic_partition_op_gpu.cu.cc | 212 OpOutputList* Tout, DoneCallback done) {
215 OP_REQUIRES_OK_ASYNC(c, c->output_list("outputs", Tout), done);
223 OP_REQUIRES_OK_ASYNC(c, Tout->allocate(p, shape, &out), done);
210 AllocateOutputs(OpKernelContext* c, const Tensor* data, const Tensor* partitions, const Tensor* partition_count, OpOutputList* Tout, DoneCallback done) argument
|