Home
last modified time | relevance | path

Searched refs:in_sizes (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dmkl_conv_grad_bias_ops.cc100 mkl_context.in_sizes[MklDims::W] = in Compute()
102 mkl_context.in_sizes[MklDims::H] = in Compute()
104 mkl_context.in_sizes[MklDims::C] = in Compute()
106 mkl_context.in_sizes[MklDims::N] = in Compute()
109 mkl_context.in_sizes[MklDims::W] = GetTensorDim(input, data_format_, 'W'); in Compute()
110 mkl_context.in_sizes[MklDims::H] = GetTensorDim(input, data_format_, 'H'); in Compute()
111 mkl_context.in_sizes[MklDims::C] = GetTensorDim(input, data_format_, 'C'); in Compute()
112 mkl_context.in_sizes[MklDims::N] = GetTensorDim(input, data_format_, 'N'); in Compute()
114 mkl_context.in_sizes); in Compute()
123 mkl_context.in_dims, mkl_context.in_sizes), in Compute()
[all …]
Dragged_range_op.cc54 std::vector<int> in_sizes; in Compute() local
55 if (!broadcast_starts) in_sizes.push_back(starts_in.shape().dim_size(0)); in Compute()
56 if (!broadcast_limits) in_sizes.push_back(limits_in.shape().dim_size(0)); in Compute()
57 if (!broadcast_deltas) in_sizes.push_back(deltas_in.shape().dim_size(0)); in Compute()
58 for (int i = 1; i < in_sizes.size(); ++i) { in Compute()
59 OP_REQUIRES(context, in_sizes[i] == in_sizes[i - 1], in Compute()
63 int64 nrows = in_sizes.empty() ? 1 : in_sizes[0]; in Compute()
Dmkl_pooling_ops_common.cc408 mkl_params->in_sizes[0] = params.tensor_in_cols; in ExtractMklOpParams()
409 mkl_params->in_sizes[1] = params.tensor_in_rows; in ExtractMklOpParams()
410 mkl_params->in_sizes[2] = params.depth; in ExtractMklOpParams()
411 mkl_params->in_sizes[3] = params.tensor_in_batch; in ExtractMklOpParams()
414 mkl_params->in_sizes); in ExtractMklOpParams()
Dmkl_conv_ops.cc564 mkl_context.in_sizes[MklDims::W] = static_cast<size_t>(input_cols); in Compute()
565 mkl_context.in_sizes[MklDims::H] = static_cast<size_t>(input_rows); in Compute()
566 mkl_context.in_sizes[MklDims::C] = static_cast<size_t>(input_depth); in Compute()
567 mkl_context.in_sizes[MklDims::N] = static_cast<size_t>(batch); in Compute()
583 mkl_context.in_sizes); 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()
687 size_t in_sizes[4]; member
710 CHECK_EQ(dnnLayoutCreate_F32(&lt_input, in_dims, in_sizes, in_strides), in MklCreateInputLayouts()
Dmkl_avgpooling_op.cc167 params.in_sizes, params.in_strides), in MklCreateLayoutsAndPrimitives()
320 mkl_context.params.in_sizes, in Compute()
391 params.in_sizes, params.in_strides), in MklCreateLayoutsAndPrimitives()
Dmkl_maxpooling_op.cc159 params.in_sizes, params.in_strides), in MklCreateLayoutsAndPrimitives()
283 mkl_context.params.in_sizes, in Compute()
325 params.in_sizes, params.in_strides), in MklCreateLayouts()
Dmkl_pooling_ops_common.h760 size_t in_sizes[4]; member
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dindex_ops_kernel_argmax_float_2d.cc32 int64* in_sizes = static_cast<int64*>(data[1]); in argmax_float_2d_xla_impl() local
36 Eigen::DSizes<Eigen::DenseIndex, 2> in_eig_sizes(in_sizes[0], in_sizes[1]); in argmax_float_2d_xla_impl()