| /external/tensorflow/tensorflow/core/kernels/ |
| tile_functor_gpu.cu.cc | 33 const int32* in_strides = buf; local
40 i_idx += t / out_strides[i] % in_dim_sizes[i] * in_strides[i];
57 gtl::InlinedVector<int32, 8> in_strides = ComputeStride<int32>(in.shape()); local
60 host_buf[i] = in_strides[i];
|
| tile_functor_cpu.cc | 31 gtl::InlinedVector<int64, 8> in_strides = ComputeStride<int64>(in.shape()); local
40 i_idx += t / out_strides[i] % in.dim_size(i) * in_strides[i];
|
| mkl_relu_op.cc | 94 mkl_context.in_strides = new size_t[mkl_context.in_dims];
97 mkl_context.in_strides[i] = mkl_context.input_shape.GetStrides()[i];
102 mkl_context.in_strides = new size_t[mkl_context.in_dims];
106 mkl_context.in_strides[0] = 1;
108 mkl_context.in_strides[i] =
109 mkl_context.in_strides[i - 1] * mkl_context.in_sizes[i - 1];
127 mkl_context.in_dims, mkl_context.in_sizes, mkl_context.in_strides);
155 size_t* in_strides; member in struct:tensorflow::MklReluOp::__anon39850
166 free(in_strides);
174 CHECK_EQ(dnnLayoutCreate_F32(<_input, in_dims, in_sizes, in_strides),
196 size_t* in_strides; member in struct:tensorflow::MklReluGradOp::__anon39851
[all...] |
| mkl_aggregate_ops.cc | 107 mkl_context.in_strides = new size_t[mkl_context.in_dims];
115 mkl_context.in_strides[i] = tmp_mkl_shape->GetStrides()[i];
122 mkl_context.in_strides[0] = 1;
124 mkl_context.in_strides[i] =
125 mkl_context.in_strides[i - 1] * mkl_context.in_sizes[i - 1];
145 mkl_context.in_dims, mkl_context.in_sizes, mkl_context.in_strides);
180 size_t* in_strides = nullptr; member in struct:tensorflow::MklAddNOp::__anon39833
190 CHECK_EQ(dnnLayoutCreate_F32(<_input1, in_dims, in_sizes, in_strides),
198 CHECK_EQ(dnnLayoutCreate_F32(<_input2, in_dims, in_sizes, in_strides),
276 delete[] in_strides;
[all...] |
| transpose_functor_gpu.cu.cc | 37 const int32* in_strides = buf; local
46 i_idx += ratio * in_strides[perm[i]];
65 gtl::InlinedVector<int32, 8> in_strides = ComputeStride<int32>(in.shape()); local
69 host_buf[i] = in_strides[i];
|
| mkl_conv_grad_bias_ops.cc | 109 GetStridesFromSizes(data_format_, mkl_context.in_strides,
147 size_t in_strides[4]; member in struct:tensorflow::MklConv2DCustomBackpropBiasOp::__anon39838
171 CHECK_EQ(dnnLayoutCreate_F32(<_input, in_dims, in_sizes, in_strides),
|
| transpose_functor_cpu.cc | 38 gtl::InlinedVector<int64, 8> in_strides = ComputeStride<int64>(in.shape()); local
42 auto transpose_fn = [=, &in_strides, &out_strides, &perm](int64 begin,
50 i_idx += ratio * in_strides[perm[i]];
|
| slice_op.cc | 343 size_t in_strides[4] = { local
362 T* ip = in_buf + (d0 * in_strides[0]);
366 T* ip1 = ip + (d1 * in_strides[1]);
371 sizeof(T) * in_strides[1]);
380 T* ip = in_buf + (d0 * in_strides[0]);
384 T* ip1 = ip + (d1 * in_strides[1]);
388 T* ip2 = ip1 + (d2 * in_strides[2]);
|
| mkl_pooling_ops_common.cc | 159 GetStridesFromSizes(data_format, mkl_params->in_strides,
|
| mkl_fused_batch_norm_op.cc | 136 mkl_context.mkl_params.in_strides);
221 size_t in_strides[4]; member in struct:tensorflow::MklFusedBatchNormOp::__anon39842::MklBatchNormParams
260 GetStridesFromSizes(tensor_format, mkl_params.in_strides,
273 mkl_params.in_sizes, mkl_params.in_strides),
423 mkl_context.mkl_params.in_strides);
499 size_t in_strides[4]; member in struct:tensorflow::MklFusedBatchNormGradOp::__anon39843::MklBatchNormParams
543 GetStridesFromSizes(tensor_format, mkl_params.in_strides,
555 mkl_params.in_sizes, mkl_params.in_strides),
566 mkl_params.in_sizes, mkl_params.in_strides),
[all...] |
| mkl_conv_grad_filter_ops.cc | 178 GetStridesFromSizes(data_format_, mkl_context.in_strides,
271 size_t in_strides[4]; member in struct:tensorflow::MklConv2DCustomBackpropFilterOp::__anon39839
305 CHECK_EQ(dnnLayoutCreate_F32(<_input, in_dims, in_sizes, in_strides),
|
| mkl_conv_grad_input_ops.cc | 181 GetStridesFromSizes(data_format, mkl_context.in_strides,
211 mkl_context.in_strides);
240 size_t in_strides[4]; member in struct:tensorflow::MklConv2DCustomBackpropInputOp::__anon39840
|
| mkl_avgpooling_op.cc | 167 params.in_sizes, params.in_strides),
321 mkl_context.params.in_strides);
391 params.in_sizes, params.in_strides),
|
| mkl_conv_ops.cc | 231 GetStridesFromSizes(data_format_, mkl_context.in_strides,
335 size_t in_strides[4]; member in struct:tensorflow::MklConv2DOp::__anon39841
357 CHECK_EQ(dnnLayoutCreate_F32(<_input, in_dims, in_sizes, in_strides),
[all...] |
| mkl_maxpooling_op.cc | 158 params.in_sizes, params.in_strides),
282 mkl_context.params.in_strides);
323 params.in_sizes, params.in_strides),
[all...] |
| mkl_lrn_op.cc | 170 size_t in_strides[4]; member in struct:tensorflow::MklLRNOp::__anon39845
405 mkl_context.in_strides[i] = mkl_context.out_strides[i] =
498 size_t in_strides[4]; member in struct:tensorflow::MklLRNGradOp::__anon39846
516 CHECK_EQ(dnnLayoutCreate_F32(<_input, in_dims, in_sizes, in_strides),
[all...] |
| mkl_pooling_ops_common.h | 406 size_t in_strides[4]; member in struct:tensorflow::__anon39849
|
