| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | runtime_fft_impl.h | 76 Eigen::DSizes<Eigen::DenseIndex, FFTRank + 1> in_dims; in EigenFftR2C() local
77 in_dims[0] = input_batch; in EigenFftR2C()
81 in_dims[i + 1] = fft_shape[i]; in EigenFftR2C()
86 input(operand, in_dims); in EigenFftR2C()
95 Eigen::Tensor<Complex, FFTRank + 1, Eigen::RowMajor> full_fft(in_dims); in EigenFftR2C()
115 Eigen::DSizes<Eigen::DenseIndex, FFTRank + 1> in_dims; in EigenFftC2R() local
116 in_dims[0] = input_batch; in EigenFftC2R()
120 in_dims[i + 1] = i == FFTRank - 1 ? fft_shape[i] / 2 + 1 : fft_shape[i]; in EigenFftC2R()
125 input(operand, in_dims); in EigenFftC2R()
138 auto neg_sizes = in_dims; in EigenFftC2R()
[all …]
|
| /external/tensorflow/tensorflow/python/keras/utils/ |
| D | conv_utils.py | 265 in_dims = len(input_shape)
267 kernel_shape = (kernel_shape,) * in_dims
269 strides = (strides,) * in_dims
273 if kernel_dims != in_dims or stride_dims != in_dims:
276 (stride_dims, in_dims, kernel_dims))
347 in_dims = len(input_shape)
349 kernel_shape = (kernel_shape,) * in_dims
351 strides = (strides,) * in_dims
355 if kernel_dims != in_dims or stride_dims != in_dims:
358 (stride_dims, in_dims, kernel_dims))
|
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorBase.h | 532 DimensionList<Index, NumDimensions> in_dims; in sum() local
533 …e>, const DimensionList<Index, NumDimensions>, const Derived>(derived(), in_dims, internal::SumRed… in sum()
544 DimensionList<Index, NumDimensions> in_dims; in mean() local
545 …e>, const DimensionList<Index, NumDimensions>, const Derived>(derived(), in_dims, internal::MeanRe… in mean()
556 DimensionList<Index, NumDimensions> in_dims; in prod() local
557 …e>, const DimensionList<Index, NumDimensions>, const Derived>(derived(), in_dims, internal::ProdRe… in prod()
568 DimensionList<Index, NumDimensions> in_dims; in maximum() local
569 …e>, const DimensionList<Index, NumDimensions>, const Derived>(derived(), in_dims, internal::MaxRed… in maximum()
580 DimensionList<Index, NumDimensions> in_dims; in minimum() local
581 …e>, const DimensionList<Index, NumDimensions>, const Derived>(derived(), in_dims, internal::MinRed… in minimum()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_transpose_op.cc | 132 memory::dims in_dims = TFShapeToMklDnnDims(in_tensor.shape()); in MKLTransposeND() local
134 memory::dims in_strides = CalculateTFStrides(in_dims); in MKLTransposeND()
140 in.SetUsrMem(in_dims, in_strides, &in_tensor); in MKLTransposeND()
143 out.SetUsrMem(in_dims, out_strides, out_tensor); in MKLTransposeND()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util.h | 1120 inline memory::dims MklDnnDimsInNCHW(const memory::dims& in_dims, in MklDnnDimsInNCHW() argument
1126 int n = in_dims[GetTensorDimIndex(format, 'N')]; in MklDnnDimsInNCHW()
1127 int c = in_dims[GetTensorDimIndex(format, 'C')]; in MklDnnDimsInNCHW()
1128 int h = in_dims[GetTensorDimIndex(format, 'H')]; in MklDnnDimsInNCHW()
1129 int w = in_dims[GetTensorDimIndex(format, 'W')]; in MklDnnDimsInNCHW()
1137 inline memory::dims MklDnnDimsInNCDHW(const memory::dims& in_dims, in MklDnnDimsInNCDHW() argument
1143 int n = in_dims[GetTensorDimIndex<3>(format, 'N')]; in MklDnnDimsInNCDHW()
1144 int c = in_dims[GetTensorDimIndex<3>(format, 'C')]; in MklDnnDimsInNCDHW()
1145 int d = in_dims[GetTensorDimIndex<3>(format, '0')]; in MklDnnDimsInNCDHW()
1146 int h = in_dims[GetTensorDimIndex<3>(format, '1')]; in MklDnnDimsInNCDHW()
[all …]
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | local.py | 801 in_dims = shape[:split_dim]
804 in_size = math_ops.reduce_prod(in_dims)
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 2010 std::vector<int64> in_dims = RandomDims(); in TEST_F() local
2012 std::uniform_int_distribution<int32> d(-1 - in_dims.size(), in_dims.size()); in TEST_F()
2015 .RandomInput(type, in_dims) in TEST_F()
2691 std::vector<int64> in_dims = RandomDims(4, 4); in TEST_F() local
2696 .RandomInput(DT_FLOAT, in_dims) in TEST_F()
2706 std::vector<int64> in_dims = RandomDims(4, 4); in TEST_F() local
2711 .RandomInput(DT_FLOAT, in_dims) in TEST_F()
2713 {in_dims[0], out_dims[0], out_dims[1], in_dims[3]}) in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier_test.cc | 4087 std::vector<int64> in_dims; in TEST_F() local
4099 in_dims.push_back(options.in_batch); in TEST_F()
4103 in_dims.push_back(options.in_height); in TEST_F()
4107 in_dims.push_back(options.in_width); in TEST_F()
4111 in_dims.push_back(options.in_channels); in TEST_F()
4144 auto in_shape = make_shape(in_dims, options.input_minor_to_major_layout); in TEST_F()
|