Home
last modified time | relevance | path

Searched refs:flat_inner_dims (Results 1 – 25 of 35) sorted by relevance

12

/external/tensorflow/tensorflow/core/framework/
Dtensor_test.cc533 auto flat_inner_dims = t.flat_inner_dims<float>(); in TEST_F() local
534 EXPECT_EQ(24, flat_inner_dims.dimension(0)); in TEST_F()
535 EXPECT_EQ(5, flat_inner_dims.dimension(1)); in TEST_F()
536 EXPECT_EQ(flat_inner_dims(0, 0), 0.01f); in TEST_F()
537 EXPECT_EQ(flat_inner_dims(23, 4), 0.02f); in TEST_F()
540 auto flat_inner_dims = t.flat_inner_dims<float, 3>(); in TEST_F() local
541 EXPECT_EQ(6, flat_inner_dims.dimension(0)); in TEST_F()
542 EXPECT_EQ(4, flat_inner_dims.dimension(1)); in TEST_F()
543 EXPECT_EQ(5, flat_inner_dims.dimension(2)); in TEST_F()
544 EXPECT_EQ(flat_inner_dims(0, 0, 0), 0.01f); in TEST_F()
[all …]
Dtensor.h474 typename TTypes<T, NDIMS>::Tensor flat_inner_dims();
579 typename TTypes<T, NDIMS>::ConstTensor flat_inner_dims() const;
913 typename TTypes<T, NDIMS>::Tensor Tensor::flat_inner_dims() { in flat_inner_dims() function
930 typename TTypes<T, NDIMS>::ConstTensor Tensor::flat_inner_dims() const { in flat_inner_dims() function
/external/tensorflow/tensorflow/core/kernels/
Dencode_proto_op.cc78 auto input_t = input.flat_inner_dims<int64>(); in TotalPackedSize()
91 auto input_t = input.flat_inner_dims<uint64>(); in TotalPackedSize()
104 auto input_t = input.flat_inner_dims<int64>(); in TotalPackedSize()
117 auto input_t = input.flat_inner_dims<int32>(); in TotalPackedSize()
155 auto input_t = input.flat_inner_dims<uint64>(); in TotalPackedSize()
168 auto input_t = input.flat_inner_dims<uint32>(); in TotalPackedSize()
181 auto input_t = input.flat_inner_dims<int32>(); in TotalPackedSize()
212 auto input_t = input.flat_inner_dims<int32>(); in TotalPackedSize()
225 auto input_t = input.flat_inner_dims<int64>(); in TotalPackedSize()
238 auto input_t = input.flat_inner_dims<int64>(); in TotalPackedSize()
[all …]
Dcross_op.cc72 in0.flat_inner_dims<Type>(); in Compute()
74 in1.flat_inner_dims<Type>(); in Compute()
76 output->flat_inner_dims<Type>(); in Compute()
Dfft_ops.cc170 auto input = Tensor(in).flat_inner_dims<complex128, FFTRank + 1>(); in DoFFT()
171 auto output = out->flat_inner_dims<complex128, FFTRank + 1>(); in DoFFT()
177 auto input = Tensor(in).flat_inner_dims<complex64, FFTRank + 1>(); in DoFFT()
178 auto output = out->flat_inner_dims<complex64, FFTRank + 1>(); in DoFFT()
213 auto input = Tensor(in).flat_inner_dims<RealT, FFTRank + 1>(); in DoRealForwardFFT()
225 auto output = out->flat_inner_dims<ComplexT, FFTRank + 1>(); in DoRealForwardFFT()
232 auto full_fft = temp.flat_inner_dims<ComplexT, FFTRank + 1>(); in DoRealForwardFFT()
247 auto input = Tensor(in).flat_inner_dims<ComplexT, FFTRank + 1>(); in DoRealBackwardFFT()
248 auto output = out->flat_inner_dims<RealT, FFTRank + 1>(); in DoRealBackwardFFT()
268 auto full_fft = temp.flat_inner_dims<ComplexT, FFTRank + 1>(); in DoRealBackwardFFT()
Dsoftmax_op.cc65 functor(context->eigen_device<Device>(), logits_in.flat_inner_dims<T>(), in Compute()
66 softmax_out->flat_inner_dims<T>(), log_); in Compute()
Disotonic_regression_op.cc179 const auto input = input_tensor.flat_inner_dims<Tin, 2>(); in Compute()
191 auto output = output_tensor->flat_inner_dims<Tout, 2>(); in Compute()
197 auto segments = segments_tensor->flat_inner_dims<int>(); in Compute()
Dfake_quant_ops.cc350 functor(context->eigen_device<Device>(), input.flat_inner_dims<float, 2>(), in Compute()
352 output->flat_inner_dims<float, 2>()); in Compute()
410 context->eigen_device<Device>(), gradient.flat_inner_dims<float, 2>(), in Compute()
411 input.flat_inner_dims<float, 2>(), min.vec<float>(), max.vec<float>(), in Compute()
412 quant_min_, quant_max_, grad_wrt_input->flat_inner_dims<float, 2>(), in Compute()
Dtopk_op.cc71 const auto& input = input_in.flat_inner_dims<T>(); in Compute()
88 auto values = values_out->flat_inner_dims<T>(); in Compute()
89 auto indices = indices_out->flat_inner_dims<int32>(); in Compute()
Dcompare_and_bitpack_op.cc66 auto input = input_t.flat_inner_dims<T>(); in Compute()
68 auto output = output_t->flat_inner_dims<uint8>(); in Compute()
Ddecode_padded_raw_op.cc73 auto out = output_tensor->flat_inner_dims<T>(); in Compute()
Ddecode_raw_op.cc75 auto out = output_tensor->flat_inner_dims<T>(); in Compute()
Dgather_nd_op.h121 auto indices_mat = indices.flat_inner_dims<Index>(); in DoGatherNd()
Dsparse_tensor_dense_add_op.cc87 const auto a_indices_mat = a_indices_t->flat_inner_dims<Index>(); in Compute()
/external/tensorflow/tensorflow/core/kernels/image/
Dcolorspace_op.cc63 typename TTypes<T, 2>::ConstTensor input_data = input.flat_inner_dims<T>(); in Compute()
64 typename TTypes<T, 2>::Tensor output_data = output->flat_inner_dims<T>(); in Compute()
100 typename TTypes<T, 2>::ConstTensor input_data = input.flat_inner_dims<T>(); in Compute()
101 typename TTypes<T, 2>::Tensor output_data = output->flat_inner_dims<T>(); in Compute()
/external/tensorflow/tensorflow/core/kernels/linalg/
Dcholesky_op.cc135 auto input_reshaped = input.template flat_inner_dims<Scalar, 3>(); in ComputeAsync()
136 auto output_reshaped = output->template flat_inner_dims<Scalar, 3>(); in ComputeAsync()
175 ->template flat_inner_dims<Scalar, 3>(); in ComputeAsync()
176 auto output_reshaped = output->template flat_inner_dims<Scalar, 3>(); in ComputeAsync()
Dsvd_op_gpu.cu.cc111 auto input_reshaped = M_copy.template flat_inner_dims<Scalar, 3>(); in RunSVD()
159 outputU_ptr = u_copy.template flat_inner_dims<Scalar, 3>().data(); in RunSVD()
161 outputV_ptr = v_copy.template flat_inner_dims<Scalar, 3>().data(); in RunSVD()
163 outputV_ptr = V->template flat_inner_dims<Scalar, 3>().data(); in RunSVD()
167 outputS_ptr = S->template flat_inner_dims<RealScalar, 2>().data(); in RunSVD()
404 auto outputU_reshaped = outputU->flat_inner_dims<Scalar, 3>(); in ComputeAsync()
408 auto outputV_reshaped = outputV->flat_inner_dims<Scalar, 3>(); in ComputeAsync()
Dqr_op_impl.h146 input.template flat_inner_dims<Scalar, 3>().dimension(0); in ComputeAsync()
205 input_transposed.flat_inner_dims<Scalar, 3>(); in ComputeAsync()
207 auto r_reshaped = r->flat_inner_dims<Scalar, 3>(); in ComputeAsync()
241 const_cast<const Tensor*>(r)->flat_inner_dims<Scalar, 3>(); in ComputeAsync()
252 auto q_reshaped = q->flat_inner_dims<Scalar, 3>(); in ComputeAsync()
Dself_adjoint_eig_v2_op_gpu.cc62 input.template flat_inner_dims<Scalar, 3>().dimension(0); in ComputeAsync()
127 auto input_copy_reshaped = input_copy.flat_inner_dims<Scalar, 3>(); in ComputeAsync()
129 eigenvalues_real.flat_inner_dims<RealScalar, 2>(); in ComputeAsync()
Ddeterminant_op.cc182 auto input_copy_reshaped = input_copy.template flat_inner_dims<Scalar, 3>(); in ComputeAsync()
200 auto output_reshaped = out->template flat_inner_dims<Scalar, 1>(); in ComputeAsync()
239 ->template flat_inner_dims<Scalar, 3>(), in ComputeAsync()
327 auto input_copy_reshaped = input_copy.template flat_inner_dims<Scalar, 3>(); in ComputeAsync()
378 ->template flat_inner_dims<Scalar, 3>(); in ComputeAsync()
Dmatrix_inverse_op.cc138 auto input_copy_reshaped = input_copy.template flat_inner_dims<Scalar, 3>(); in ComputeAsync()
164 auto output_reshaped = output->template flat_inner_dims<Scalar, 3>(); in ComputeAsync()
Dmatrix_band_part_op.cc60 auto input_reshaped = input.flat_inner_dims<T, 3>(); in Compute()
103 auto output_reshaped = output->flat_inner_dims<T, 3>(); in Compute()
Dmatrix_set_diag_op.cc157 auto input_reshaped = input.flat_inner_dims<T, 3>(); in Compute()
162 auto output_reshaped = output->flat_inner_dims<T, 3>(); in Compute()
Dmatrix_solve_op.cc199 auto input_copy_reshaped = input_copy.template flat_inner_dims<Scalar, 3>(); in ComputeAsync()
284 transposed_rhs.template flat_inner_dims<Scalar, 3>(); in ComputeAsync()
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_requantization_range_per_channel_op.cc66 auto input_matrix = input.flat_inner_dims<qint32>(); in Compute()

12