| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| runtime_fft_impl.h | 70 Eigen::DSizes<Eigen::DenseIndex, FFTRank + 1> in_dims; local
71 in_dims[0] = input_batch;
76 in_dims[i + 1] = fft_shape[i];
82 input(operand, in_dims);
111 Eigen::DSizes<Eigen::DenseIndex, FFTRank + 1> in_dims; local
112 in_dims[0] = input_batch;
117 in_dims[i + 1] = i == FFTRank - 1 ? fft_shape[i] / 2 + 1 : fft_shape[i];
123 input(operand, in_dims);
137 auto neg_sizes = in_dims;
138 neg_sizes[FFTRank] = fft_shape[FFTRank - 1] - in_dims[FFTRank]
[all...] |
| /external/tensorflow/tensorflow/core/kernels/ |
| mkl_relu_op.cc | 92 mkl_context.in_dims = mkl_context.input_shape.GetDimension();
93 mkl_context.in_sizes = new size_t[mkl_context.in_dims];
94 mkl_context.in_strides = new size_t[mkl_context.in_dims];
95 for (int i = 0; i < mkl_context.in_dims; i++) {
100 mkl_context.in_dims = input.dims();
101 mkl_context.in_sizes = new size_t[mkl_context.in_dims];
102 mkl_context.in_strides = new size_t[mkl_context.in_dims];
103 for (int i = 0; i < mkl_context.in_dims; i++) {
104 mkl_context.in_sizes[i] = input.dim_size((mkl_context.in_dims - 1) - i);
107 for (int i = 1; i < mkl_context.in_dims; i++)
153 int in_dims; member in struct:tensorflow::MklReluOp::__anon39850
194 int in_dims; member in struct:tensorflow::MklReluGradOp::__anon39851
[all...] |
| mkl_aggregate_ops.cc | 87 mkl_context.in_dims = input1_in_mkl_format
90 mkl_context.in_dims = input2_in_mkl_format
106 mkl_context.in_sizes = new size_t[mkl_context.in_dims];
107 mkl_context.in_strides = new size_t[mkl_context.in_dims];
113 for (int i = 0; i < mkl_context.in_dims; i++) {
118 for (int i = 0; i < mkl_context.in_dims; i++) {
120 input0.dim_size((mkl_context.in_dims - 1) - i);
123 for (int i = 1; i < mkl_context.in_dims; i++) {
145 mkl_context.in_dims, mkl_context.in_sizes, mkl_context.in_strides);
148 mkl_context.in_dims, mkl_context.input1_shape.GetTfToMklDimMap())
178 int in_dims; member in struct:tensorflow::MklAddNOp::__anon39833
[all...] |
| mkl_conv_grad_bias_ops.cc | 93 mkl_context.in_dims = 4;
119 mkl_context.in_dims, mkl_context.in_sizes),
144 int in_dims; member in struct:tensorflow::MklConv2DCustomBackpropBiasOp::__anon39838
171 CHECK_EQ(dnnLayoutCreate_F32(<_input, in_dims, in_sizes, in_strides),
|
| mkl_conv_grad_input_ops.cc | 155 mkl_context.in_dims = 4;
200 mkl_context.in_dims, mkl_context.in_sizes, mkl_context.out_sizes,
210 mklOutputShape.SetTfLayout(mkl_context.in_dims, mkl_context.in_sizes,
214 mklOutputShape.SetTfDimOrder(mkl_context.in_dims, data_format);
238 int in_dims; member in struct:tensorflow::MklConv2DCustomBackpropInputOp::__anon39840
259 CHECK_EQ(dnnLayoutCreate_F32(<_filter, in_dims, filter_size,
267 CHECK_EQ(dnnLayoutCreate_F32(<_outbackprop, in_dims, out_sizes,
|
| mkl_lrn_op.cc | 102 mkl_context.in_dims = input_in_mkl_format
105 OP_REQUIRES(context, mkl_context.in_dims == 4,
120 if (mkl_context.input_shape.tf_dim_idx(mkl_context.in_dims - 1) ==
168 size_t in_dims; member in struct:tensorflow::MklLRNOp::__anon39845
221 mkl_output_mkl_shape.SetTfLayout(in_dims, input_shape.GetSizes(),
223 mkl_output_mkl_shape.SetTfDimOrder(in_dims,
239 mkl_workspace_mkl_shape.SetTfLayout(in_dims, input_shape.GetSizes(),
241 mkl_workspace_mkl_shape.SetTfDimOrder(in_dims,
382 mkl_context.in_dims = inimage_in_mkl_format
385 OP_REQUIRES(context, mkl_context.in_dims == 4
496 size_t in_dims; member in struct:tensorflow::MklLRNGradOp::__anon39846
[all...] |
| mkl_conv_ops.cc | 208 mkl_context.in_dims = input_in_mkl_format
260 mkl_context.in_dims, mkl_context.in_sizes, mkl_context.out_sizes,
268 mkl_context.in_dims, mkl_context.in_sizes, mkl_context.out_sizes,
278 mkl_output_mkl_shape.SetTfLayout(mkl_context.in_dims, mkl_context.out_sizes,
282 mkl_output_mkl_shape.SetTfDimOrder(mkl_context.in_dims, data_format_);
333 int in_dims; member in struct:tensorflow::MklConv2DOp::__anon39841
357 CHECK_EQ(dnnLayoutCreate_F32(<_input, in_dims, in_sizes, in_strides),
[all...] |
| mkl_conv_grad_filter_ops.cc | 153 mkl_context.in_dims = input_in_mkl_format
209 dnnAlgorithmConvolutionDirect, mkl_context.in_dims,
269 int in_dims; member in struct:tensorflow::MklConv2DCustomBackpropFilterOp::__anon39839
305 CHECK_EQ(dnnLayoutCreate_F32(<_input, in_dims, in_sizes, in_strides),
|
| mkl_fused_batch_norm_op.cc | 421 mkl_shape_in_backprop.SetTfLayout(mkl_context.mkl_params.in_dims,
424 mkl_shape_in_backprop.SetTfDimOrder(mkl_context.mkl_params.in_dims,
497 size_t in_dims; member in struct:tensorflow::MklFusedBatchNormGradOp::__anon39843::MklBatchNormParams
527 mkl_params.in_dims = input_in_mkl_format
554 dnnLayoutCreate_F32(&mkl_lt_input, mkl_params.in_dims,
565 dnnLayoutCreate_F32(&mkl_lt_out_backprop, mkl_params.in_dims,
[all...] |
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| TensorBase.h | 532 DimensionList<Index, NumDimensions> in_dims; local
533 return TensorReductionOp<internal::SumReducer<CoeffReturnType>, const DimensionList<Index, NumDimensions>, const Derived>(derived(), in_dims, internal::SumReducer<CoeffReturnType>());
544 DimensionList<Index, NumDimensions> in_dims; local
545 return TensorReductionOp<internal::MeanReducer<CoeffReturnType>, const DimensionList<Index, NumDimensions>, const Derived>(derived(), in_dims, internal::MeanReducer<CoeffReturnType>());
556 DimensionList<Index, NumDimensions> in_dims; local
557 return TensorReductionOp<internal::ProdReducer<CoeffReturnType>, const DimensionList<Index, NumDimensions>, const Derived>(derived(), in_dims, internal::ProdReducer<CoeffReturnType>());
568 DimensionList<Index, NumDimensions> in_dims; local
569 return TensorReductionOp<internal::MaxReducer<CoeffReturnType>, const DimensionList<Index, NumDimensions>, const Derived>(derived(), in_dims, internal::MaxReducer<CoeffReturnType>());
580 DimensionList<Index, NumDimensions> in_dims; local
581 return TensorReductionOp<internal::MinReducer<CoeffReturnType>, const DimensionList<Index, NumDimensions>, const Derived>(derived(), in_dims, internal::MinReducer<CoeffReturnType>())
593 DimensionList<Index, NumDimensions> in_dims; local
606 DimensionList<Index, NumDimensions> in_dims; local
615 array<Index, NumDimensions> in_dims; local
628 array<Index, NumDimensions> in_dims; local
641 array<Index, 1> in_dims; local
654 array<Index, 1> in_dims; local
[all...] |
| /external/tensorflow/tensorflow/core/util/ |
| mkl_util.h | [all...] |
| /external/tensorflow/tensorflow/compiler/tests/ |
| randomized_tests.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| algebraic_simplifier_test.cc | [all...] |
