/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
D | runtime_fft_impl.h | 68 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/ |
D | conv_utils.py | 278 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/ |
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/ |
D | mkl_transpose_op.cc | 131 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()
|
D | mkl_conv_grad_bias_ops.cc | 97 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(<_input, in_dims, in_sizes, in_strides), in MklCreateInputLayouts()
|
D | mkl_conv_ops.cc | 559 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(<_input, in_dims, in_sizes, in_strides), in MklCreateInputLayouts()
|
/external/tensorflow/tensorflow/python/keras/layers/ |
D | local.py | 723 in_dims = shape[:split_dim] 726 in_size = K.math_ops.reduce_prod(in_dims)
|
/external/tensorflow/tensorflow/compiler/tests/ |
D | randomized_tests.cc | 2009 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/ |
D | mkl_util.h | 1549 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/ |
D | algebraic_simplifier_test.cc | 3323 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()
|