| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| index_ops_kernel_argmax_float_2d.cc | 33 int64* out_sizes = static_cast<int64*>(data[2]); local
40 Eigen::DSizes<Eigen::DenseIndex, 1> out_eig_sizes(out_sizes[0]);
|
| /external/tensorflow/tensorflow/core/kernels/ |
| extract_image_patches_op.cc | 95 const std::vector<int64> out_sizes = {batch, out_rows, out_cols, variable
97 TensorShape out_shape(out_sizes);
|
| extract_volume_patches_op.cc | 121 const std::vector<int64> out_sizes = { variable
124 TensorShape out_shape(out_sizes);
|
| dilation_ops.cc | 138 const std::vector<int64> out_sizes = {batch, out_rows, out_cols, depth}; variable
139 TensorShape out_shape(out_sizes);
|
| mkl_conv_grad_bias_ops.cc | 117 mkl_context.out_sizes[0] = mkl_context.c_size;
152 size_t out_sizes[1]; member in struct:tensorflow::MklConv2DCustomBackpropBiasOp::__anon45424
170 CHECK_EQ(dnnLayoutCreate_F32(<_outbackprop, 1, out_sizes, out_strides),
|
| mkl_pooling_ops_common.h | 762 size_t out_sizes[4]; member in struct:tensorflow::__anon45429
|
| mkl_conv_ops.cc | 569 mkl_context.out_sizes[MklDims::W] = static_cast<size_t>(out_cols);
570 mkl_context.out_sizes[MklDims::H] = static_cast<size_t>(out_rows);
571 mkl_context.out_sizes[MklDims::C] = static_cast<size_t>(out_depth);
572 mkl_context.out_sizes[MklDims::N] = static_cast<size_t>(batch);
581 mkl_context.out_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,
629 mkl_output_mkl_shape.SetTfLayout(mkl_context.in_dims, mkl_context.out_sizes,
689 size_t out_sizes[4]; member in struct:tensorflow::MklConvOp::__anon45425
[all...] |
