Home
last modified time | relevance | path

Searched refs:out_sizes (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/python/kernel_tests/
Dneon_depthwise_conv_op_test.py43 out_sizes = [[4, 5, 5, 96], [4, 8, 8, 84], [4, 17, 17, 192], [4, 35, 35, 2],
51 for i, f, o, s, p in zip(input_sizes, filter_sizes, out_sizes, strides,
70 out_sizes = [[2, 5, 8, 2], [4, 2, 2, 2], [2, 4, 4, 4], [1, 15, 15, 2],
78 for i, f, o, s, p in zip(input_sizes, filter_sizes, out_sizes, strides,
Ddepthwise_conv_op_test.py50 out_sizes = [[4, 5, 5, 96], [4, 8, 8, 84], [4, 17, 17, 192], [4, 9, 27, 8],
59 for i, f, o, s, p in zip(input_sizes, filter_sizes, out_sizes, strides,
78 out_sizes = [[2, 5, 8, 2], [4, 2, 2, 2], [2, 4, 4, 4], [1, 15, 15, 2],
86 for i, f, o, s, p in zip(input_sizes, filter_sizes, out_sizes, strides,
Dconv_ops_test.py102 out_sizes = [[4, 5, 5, 128], [4, 8, 8, 384], [4, 8, 8, 384],
132 for o in out_sizes:
145 for i, f, o, s, p in zip(input_sizes, filter_sizes, out_sizes, strides,
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dindex_ops_kernel_argmax_float_2d.cc33 int64* out_sizes = static_cast<int64*>(data[2]); in argmax_float_2d_xla_impl() local
40 Eigen::DSizes<Eigen::DenseIndex, 1> out_eig_sizes(out_sizes[0]); in argmax_float_2d_xla_impl()
/external/tensorflow/tensorflow/core/kernels/
Dmkl_conv_grad_bias_ops.cc117 mkl_context.out_sizes[0] = mkl_context.c_size; in Compute()
152 size_t out_sizes[1]; member
170 CHECK_EQ(dnnLayoutCreate_F32(&lt_outbackprop, 1, out_sizes, out_strides), in MklCreateInputLayouts()
Dmkl_pooling_ops_common.cc416 mkl_params->out_sizes[0] = params.out_width; in ExtractMklOpParams()
417 mkl_params->out_sizes[1] = params.out_height; in ExtractMklOpParams()
418 mkl_params->out_sizes[2] = params.depth; in ExtractMklOpParams()
419 mkl_params->out_sizes[3] = params.tensor_in_batch; in ExtractMklOpParams()
422 mkl_params->out_sizes); in ExtractMklOpParams()
Dextract_image_patches_op.cc95 const std::vector<int64> out_sizes = {batch, out_rows, out_cols, in Compute() local
97 TensorShape out_shape(out_sizes); in Compute()
Dextract_volume_patches_op.cc121 const std::vector<int64> out_sizes = { in Compute() local
124 TensorShape out_shape(out_sizes); in Compute()
Dmkl_conv_ops.cc569 mkl_context.out_sizes[MklDims::W] = static_cast<size_t>(out_cols); in Compute()
570 mkl_context.out_sizes[MklDims::H] = static_cast<size_t>(out_rows); in Compute()
571 mkl_context.out_sizes[MklDims::C] = static_cast<size_t>(out_depth); in Compute()
572 mkl_context.out_sizes[MklDims::N] = static_cast<size_t>(batch); in Compute()
581 mkl_context.out_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()
629 mkl_output_mkl_shape.SetTfLayout(mkl_context.in_dims, mkl_context.out_sizes, in Compute()
689 size_t out_sizes[4]; member
Ddilation_ops.cc138 const std::vector<int64> out_sizes = {batch, out_rows, out_cols, depth}; in Compute() local
139 TensorShape out_shape(out_sizes); in Compute()
Dmkl_avgpooling_op.cc129 mkl_context.params.out_sizes, in Compute()
382 params.out_sizes, params.out_strides), in MklCreateLayoutsAndPrimitives()
Dmkl_maxpooling_op.cc110 mkl_context.params.out_sizes, in Compute()
335 params.out_sizes, params.out_strides), in MklCreateLayouts()
Dmkl_pooling_ops_common.h762 size_t out_sizes[4]; member
/external/tensorflow/tensorflow/compiler/tests/
Ddepthwise_conv_op_test.py76 out_sizes = [[4, 5, 5, 96], [4, 8, 8, 84], [4, 17, 17, 192], [4, 9, 27, 8],
85 for i, f, o, s, p in zip(input_sizes, filter_sizes, out_sizes, strides,
104 out_sizes = [[2, 5, 8, 2], [4, 2, 2, 2], [2, 4, 4, 4], [1, 15, 15, 2],
112 for i, f, o, s, p in zip(input_sizes, filter_sizes, out_sizes, strides,
/external/tensorflow/tensorflow/contrib/fused_conv/python/ops/
Dfused_conv2d_bias_activation_op_test_base.py107 out_sizes = [[4, 5, 5, 128], [4, 8, 8, 384], [4, 8, 8, 384], [4, 8, 8, 192], [
139 for o in out_sizes:
152 for i, f, o, s, p in zip(input_sizes, filter_sizes, out_sizes, strides,
/external/deqp/external/openglcts/modules/gl/
Dgl3cTextureSwizzleTests.hpp256 void prepareSourceTexture(size_t format_idx, size_t target_idx, glw::GLint out_sizes[4]);
Dgl3cTextureSwizzleTests.cpp3450 void SmokeTest::prepareSourceTexture(size_t format_idx, size_t target_idx, glw::GLint out_sizes[4]) in prepareSourceTexture()
3576 gl.getTexLevelParameteriv(target_get_prm, 0 /* level */, GL_TEXTURE_STENCIL_SIZE, out_sizes + 1); in prepareSourceTexture()
3586 gl.getTexLevelParameteriv(target_get_prm, 0 /* level */, GL_TEXTURE_DEPTH_SIZE, out_sizes + 0); in prepareSourceTexture()
3593 gl.getTexLevelParameteriv(target_get_prm, 0 /* level */, GL_TEXTURE_RED_SIZE, out_sizes + 0); in prepareSourceTexture()
3596 gl.getTexLevelParameteriv(target_get_prm, 0 /* level */, GL_TEXTURE_GREEN_SIZE, out_sizes + 1); in prepareSourceTexture()
3599 gl.getTexLevelParameteriv(target_get_prm, 0 /* level */, GL_TEXTURE_BLUE_SIZE, out_sizes + 2); in prepareSourceTexture()
3602 gl.getTexLevelParameteriv(target_get_prm, 0 /* level */, GL_TEXTURE_ALPHA_SIZE, out_sizes + 3); in prepareSourceTexture()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dreference_ops.h3953 int out_sizes[4]; in Transpose() local
3957 out_sizes[k] = MatchingDim(input_shape, extended_perm[k], output_shape, k); in Transpose()
3963 for (o[3] = 0; o[3] < out_sizes[3]; o[3]++) { in Transpose()
3965 for (o[2] = 0; o[2] < out_sizes[2]; o[2]++) { in Transpose()
3967 for (o[1] = 0; o[1] < out_sizes[1]; o[1]++) { in Transpose()
3969 for (o[0] = 0; o[0] < out_sizes[0]; o[0]++) { in Transpose()