| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| index_ops_kernel_argmax_float_2d.cc | 32 int64* in_sizes = static_cast<int64*>(data[1]); local
36 Eigen::DSizes<Eigen::DenseIndex, 2> in_eig_sizes(in_sizes[0], in_sizes[1]);
|
| /external/tensorflow/tensorflow/core/kernels/ |
| ragged_range_op.cc | 54 std::vector<int> in_sizes; variable
55 if (!broadcast_starts) in_sizes.push_back(starts_in.shape().dim_size(0));
56 if (!broadcast_limits) in_sizes.push_back(limits_in.shape().dim_size(0));
57 if (!broadcast_deltas) in_sizes.push_back(deltas_in.shape().dim_size(0));
58 for (int i = 1; i < in_sizes.size(); ++i) {
59 OP_REQUIRES(context, in_sizes[i] == in_sizes[i - 1],
63 int64 nrows = in_sizes.empty() ? 1 : in_sizes[0];
|
| mkl_conv_grad_bias_ops.cc | 100 mkl_context.in_sizes[MklDims::W] =
102 mkl_context.in_sizes[MklDims::H] =
104 mkl_context.in_sizes[MklDims::C] =
106 mkl_context.in_sizes[MklDims::N] =
109 mkl_context.in_sizes[MklDims::W] = GetTensorDim(input, data_format_, 'W');
110 mkl_context.in_sizes[MklDims::H] = GetTensorDim(input, data_format_, 'H');
111 mkl_context.in_sizes[MklDims::C] = GetTensorDim(input, data_format_, 'C');
112 mkl_context.in_sizes[MklDims::N] = GetTensorDim(input, data_format_, 'N');
114 mkl_context.in_sizes);
123 mkl_context.in_dims, mkl_context.in_sizes),
150 size_t in_sizes[4]; member in struct:tensorflow::MklConv2DCustomBackpropBiasOp::__anon45424
[all...] |
| mkl_pooling_ops_common.h | 760 size_t in_sizes[4]; member in struct:tensorflow::__anon45429
|
| mkl_conv_ops.cc | 564 mkl_context.in_sizes[MklDims::W] = static_cast<size_t>(input_cols);
565 mkl_context.in_sizes[MklDims::H] = static_cast<size_t>(input_rows);
566 mkl_context.in_sizes[MklDims::C] = static_cast<size_t>(input_depth);
567 mkl_context.in_sizes[MklDims::N] = static_cast<size_t>(batch);
583 mkl_context.in_sizes);
611 mkl_context.in_dims, mkl_context.in_sizes, mkl_context.out_sizes,
619 mkl_context.in_dims, mkl_context.in_sizes, mkl_context.out_sizes,
687 size_t in_sizes[4]; member in struct:tensorflow::MklConvOp::__anon45425
710 CHECK_EQ(dnnLayoutCreate_F32(<_input, in_dims, in_sizes, in_strides),
[all...] |
