/external/tensorflow/tensorflow/core/framework/ |
D | tensor_test.cc | 500 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/ |
D | encode_proto_op.cc | 78 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 …]
|
D | colorspace_op.cc | 65 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()
|
D | cross_op.cc | 72 in0.flat_inner_dims<Type>(); in Compute() 74 in1.flat_inner_dims<Type>(); in Compute() 76 output->flat_inner_dims<Type>(); in Compute()
|
D | fft_ops.cc | 137 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()
|
D | qr_op_impl.h | 146 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()
|
D | matrix_diag_op.cc | 72 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()
|
D | self_adjoint_eig_v2_op_gpu.cc | 62 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()
|
D | matrix_set_diag_op.cc | 81 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()
|
D | determinant_op.cc | 182 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()
|
D | softmax_op.cc | 72 functor(context->eigen_device<Device>(), logits_in.flat_inner_dims<T>(), in Compute() 73 softmax_out->flat_inner_dims<T>(), log_); in Compute()
|
D | fake_quant_ops.cc | 336 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()
|
D | svd_op_gpu.cu.cc | 126 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()
|
D | topk_op.cc | 71 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()
|
D | cholesky_op.cc | 136 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()
|
D | mkl_batch_matmul_op.cc | 102 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()
|
D | compare_and_bitpack_op.cc | 66 auto input = input_t.flat_inner_dims<T>(); in Compute() 68 auto output = output_t->flat_inner_dims<uint8>(); in Compute()
|
D | matrix_inverse_op.cc | 138 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()
|
D | matrix_band_part_op.cc | 59 auto input_reshaped = input.flat_inner_dims<T, 3>(); in Compute() 102 auto output_reshaped = output->flat_inner_dims<T, 3>(); in Compute()
|
D | matrix_solve_op.cc | 199 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()
|
D | decode_raw_op.cc | 70 auto out = output_tensor->flat_inner_dims<T>(); in Compute()
|
D | mkl_requantization_range_per_channel_op.cc | 64 auto input_matrix = input.flat_inner_dims<qint32>(); in Compute()
|
/external/tensorflow/tensorflow/contrib/coder/kernels/ |
D | pmf_to_cdf_op_test.cc | 70 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()
|
D | pmf_to_cdf_op.cc | 62 auto pmf = pmf_tensor.flat_inner_dims<float, 2>(); in Compute() 63 auto cdf = cdf_tensor->flat_inner_dims<int32, 2>(); in Compute()
|
D | range_coder_ops.cc | 163 cdf.flat_inner_dims<int32, 2>(), cdf_shape, output); \ in Compute() 255 cdf.flat_inner_dims<int32>(), cdf_shape, encoded); \ in Compute()
|