Home
last modified time | relevance | path

Searched refs:in_dims (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Druntime_fft_impl.h68 Eigen::DSizes<Eigen::DenseIndex, FFTRank + 1> in_dims; in EigenFftR2C() local
69 in_dims[0] = input_batch; in EigenFftR2C()
73 in_dims[i + 1] = fft_shape[i]; in EigenFftR2C()
78 input(operand, in_dims); in EigenFftR2C()
87 Eigen::Tensor<complex64, FFTRank + 1, Eigen::RowMajor> full_fft(in_dims); in EigenFftR2C()
107 Eigen::DSizes<Eigen::DenseIndex, FFTRank + 1> in_dims; in EigenFftC2R() local
108 in_dims[0] = input_batch; in EigenFftC2R()
112 in_dims[i + 1] = i == FFTRank - 1 ? fft_shape[i] / 2 + 1 : fft_shape[i]; in EigenFftC2R()
117 input(operand, in_dims); in EigenFftC2R()
130 auto neg_sizes = in_dims; in EigenFftC2R()
[all …]
/external/tensorflow/tensorflow/python/keras/utils/
Dconv_utils.py278 in_dims = len(input_shape)
280 kernel_shape = (kernel_shape,) * in_dims
282 strides = (strides,) * in_dims
286 if kernel_dims != in_dims or stride_dims != in_dims:
289 (stride_dims, in_dims, kernel_dims))
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorBase.h532 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/
Dmkl_transpose_op.cc131 memory::dims in_dims = TFShapeToMklDnnDims(in_tensor.shape()); in MKLTransposeND() local
133 memory::dims in_strides = CalculateTFStrides(in_dims); in MKLTransposeND()
138 in.SetUsrMem(in_dims, in_strides, &in_tensor); in MKLTransposeND()
141 out.SetUsrMem(in_dims, out_strides, out_tensor); in MKLTransposeND()
Dmkl_conv_grad_bias_ops.cc97 mkl_context.in_dims = 4; in Compute()
123 mkl_context.in_dims, mkl_context.in_sizes), in Compute()
148 int in_dims; member
175 CHECK_EQ(dnnLayoutCreate_F32(&lt_input, in_dims, in_sizes, in_strides), in MklCreateInputLayouts()
Dmkl_conv_ops.cc559 mkl_context.in_dims = input_in_mkl_format in Compute()
611 mkl_context.in_dims, mkl_context.in_sizes, mkl_context.out_sizes, in Compute()
619 mkl_context.in_dims, mkl_context.in_sizes, mkl_context.out_sizes, in Compute()
629 mkl_output_mkl_shape.SetTfLayout(mkl_context.in_dims, mkl_context.out_sizes, in Compute()
633 mkl_output_mkl_shape.SetTfDimOrder(mkl_context.in_dims, data_format_); in Compute()
686 int in_dims; member
710 CHECK_EQ(dnnLayoutCreate_F32(&lt_input, in_dims, in_sizes, in_strides), in MklCreateInputLayouts()
/external/tensorflow/tensorflow/python/keras/layers/
Dlocal.py723 in_dims = shape[:split_dim]
726 in_size = K.math_ops.reduce_prod(in_dims)
/external/tensorflow/tensorflow/compiler/tests/
Drandomized_tests.cc2009 std::vector<int64> in_dims = RandomDims(); in TEST_F() local
2011 std::uniform_int_distribution<int32> d(-1 - in_dims.size(), in_dims.size()); in TEST_F()
2014 .RandomInput(type, in_dims) in TEST_F()
2690 std::vector<int64> in_dims = RandomDims(4, 4); in TEST_F() local
2695 .RandomInput(DT_FLOAT, in_dims) in TEST_F()
2705 std::vector<int64> in_dims = RandomDims(4, 4); in TEST_F() local
2710 .RandomInput(DT_FLOAT, in_dims) in TEST_F()
2712 {in_dims[0], out_dims[0], out_dims[1], in_dims[3]}) in TEST_F()
/external/tensorflow/tensorflow/core/util/
Dmkl_util.h1549 inline memory::dims MklDnnDimsInNCHW(const memory::dims& in_dims, in MklDnnDimsInNCHW() argument
1555 int n = in_dims[GetTensorDimIndex(format, 'N')]; in MklDnnDimsInNCHW()
1556 int c = in_dims[GetTensorDimIndex(format, 'C')]; in MklDnnDimsInNCHW()
1557 int h = in_dims[GetTensorDimIndex(format, 'H')]; in MklDnnDimsInNCHW()
1558 int w = in_dims[GetTensorDimIndex(format, 'W')]; in MklDnnDimsInNCHW()
/external/tensorflow/tensorflow/compiler/xla/service/
Dalgebraic_simplifier_test.cc3323 std::vector<int64> in_dims; in TEST_F() local
3335 in_dims.push_back(options.in_batch); in TEST_F()
3339 in_dims.push_back(options.in_height); in TEST_F()
3343 in_dims.push_back(options.in_width); in TEST_F()
3347 in_dims.push_back(options.in_channels); in TEST_F()
3372 auto out_dims = in_dims; in TEST_F()
3383 auto in_shape = make_shape(in_dims, options.input_minor_to_major_layout); in TEST_F()