Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/framework/
Dtensor_test.cc500 auto flat_inner_dims = t.flat_inner_dims<float>(); in TEST_F() local
501 EXPECT_EQ(24, flat_inner_dims.dimension(0)); in TEST_F()
502 EXPECT_EQ(5, flat_inner_dims.dimension(1)); in TEST_F()
503 EXPECT_EQ(flat_inner_dims(0, 0), 0.01f); in TEST_F()
504 EXPECT_EQ(flat_inner_dims(23, 4), 0.02f); in TEST_F()
507 auto flat_inner_dims = t.flat_inner_dims<float, 3>(); in TEST_F() local
508 EXPECT_EQ(6, flat_inner_dims.dimension(0)); in TEST_F()
509 EXPECT_EQ(4, flat_inner_dims.dimension(1)); in TEST_F()
510 EXPECT_EQ(5, flat_inner_dims.dimension(2)); in TEST_F()
511 EXPECT_EQ(flat_inner_dims(0, 0, 0), 0.01f); in TEST_F()
[all …]
/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 …]
Dcolorspace_op.cc65 typename TTypes<T, 2>::ConstTensor input_data = input.flat_inner_dims<T>(); in Compute()
66 typename TTypes<T, 2>::Tensor output_data = output->flat_inner_dims<T>(); in Compute()
102 typename TTypes<T, 2>::ConstTensor input_data = input.flat_inner_dims<T>(); in Compute()
103 typename TTypes<T, 2>::Tensor output_data = output->flat_inner_dims<T>(); in Compute()
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.cc137 auto input = Tensor(in).flat_inner_dims<complex64, FFTRank + 1>(); in DoFFT()
138 auto output = out->flat_inner_dims<complex64, FFTRank + 1>(); in DoFFT()
144 auto input = Tensor(in).flat_inner_dims<complex128, FFTRank + 1>(); in DoFFT()
145 auto output = out->flat_inner_dims<complex128, FFTRank + 1>(); in DoFFT()
151 auto input = Tensor(in).flat_inner_dims<float, FFTRank + 1>(); in DoFFT()
163 auto output = out->flat_inner_dims<complex64, FFTRank + 1>(); in DoFFT()
170 auto full_fft = temp.flat_inner_dims<complex64, FFTRank + 1>(); in DoFFT()
180 auto input = Tensor(in).flat_inner_dims<complex64, FFTRank + 1>(); in DoFFT()
181 auto output = out->flat_inner_dims<float, FFTRank + 1>(); in DoFFT()
201 auto full_fft = temp.flat_inner_dims<complex64, FFTRank + 1>(); in DoFFT()
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()
Dmatrix_diag_op.cc72 auto output_reshaped = output->flat_inner_dims<T, 2>(); in Compute()
73 auto input_reshaped = input.flat_inner_dims<T, 3>(); in Compute()
101 auto input_reshaped = input.flat_inner_dims<T, 2>(); in Compute()
109 auto output_reshaped = output->flat_inner_dims<T, 3>(); in Compute()
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()
Dmatrix_set_diag_op.cc81 auto input_reshaped = input.flat_inner_dims<T, 3>(); in Compute()
82 auto diag_reshaped = diag.flat_inner_dims<T, 2>(); in Compute()
86 auto output_reshaped = output->flat_inner_dims<T, 3>(); in Compute()
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()
Dsoftmax_op.cc72 functor(context->eigen_device<Device>(), logits_in.flat_inner_dims<T>(), in Compute()
73 softmax_out->flat_inner_dims<T>(), log_); in Compute()
Dfake_quant_ops.cc336 functor(context->eigen_device<Device>(), input.flat_inner_dims<float, 2>(), in Compute()
338 output->flat_inner_dims<float, 2>()); in Compute()
396 context->eigen_device<Device>(), gradient.flat_inner_dims<float, 2>(), in Compute()
397 input.flat_inner_dims<float, 2>(), min.vec<float>(), max.vec<float>(), in Compute()
398 quant_min_, quant_max_, grad_wrt_input->flat_inner_dims<float, 2>(), in Compute()
Dsvd_op_gpu.cu.cc126 auto input_reshaped = M_copy.template flat_inner_dims<Scalar, 3>(); in RunSVD()
128 outputS_ptr = S->template flat_inner_dims<RealScalar, 2>().data(); in RunSVD()
130 outputU_ptr = u_copy.template flat_inner_dims<Scalar, 3>().data(); in RunSVD()
131 outputV_ptr = V->template flat_inner_dims<Scalar, 3>().data(); in RunSVD()
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()
Dcholesky_op.cc136 auto input_reshaped = input.template flat_inner_dims<Scalar, 3>(); in ComputeAsync()
137 auto output_reshaped = output->template flat_inner_dims<Scalar, 3>(); in ComputeAsync()
Dmkl_batch_matmul_op.cc102 auto rhs_reshaped = rhs.template flat_inner_dims<Scalar, 3>(); in Compute()
103 auto lhs_reshaped = lhs.template flat_inner_dims<Scalar, 3>(); in Compute()
104 auto out_reshaped = out->template flat_inner_dims<Scalar, 3>(); 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()
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.cc59 auto input_reshaped = input.flat_inner_dims<T, 3>(); in Compute()
102 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()
Ddecode_raw_op.cc70 auto out = output_tensor->flat_inner_dims<T>(); in Compute()
Dmkl_requantization_range_per_channel_op.cc64 auto input_matrix = input.flat_inner_dims<qint32>(); in Compute()
/external/tensorflow/tensorflow/contrib/coder/kernels/
Dpmf_to_cdf_op_test.cc70 auto pmf = pmf_tensor.flat_inner_dims<float, 2>(); in Verify()
71 auto cdf = cdf_tensor.flat_inner_dims<int32, 2>(); in Verify()
92 auto matrix = pmf.flat_inner_dims<float, 2>(); in TEST_F()
112 auto matrix = pmf.flat_inner_dims<float, 2>(); in TEST_F()
Dpmf_to_cdf_op.cc62 auto pmf = pmf_tensor.flat_inner_dims<float, 2>(); in Compute()
63 auto cdf = cdf_tensor->flat_inner_dims<int32, 2>(); in Compute()
Drange_coder_ops.cc163 cdf.flat_inner_dims<int32, 2>(), cdf_shape, output); \ in Compute()
255 cdf.flat_inner_dims<int32>(), cdf_shape, encoded); \ in Compute()

12