Home
last modified time | relevance | path

Searched refs:depth_index (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/util/
Dpadding.cc53 const int32 depth_index = GetTensorFeatureDimIndex(num_dims, data_format); in CheckValidPadding() local
56 explicit_paddings[2 * depth_index] != 0 || in CheckValidPadding()
57 explicit_paddings[2 * depth_index + 1] != 0) { in CheckValidPadding()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Ddepthwiseconv_uint8_transitional.h2562 for (int depth_index = 0; depth_index < 4; ++depth_index) {
2563 memcpy(selected_data[k_height][depth_index],
2564 &left_bank[k_height][depth_index][offset], 4 - offset);
2565 memcpy(&selected_data[k_height][depth_index][4 - offset],
2566 right_bank[k_height][depth_index], offset);
2997 for (int depth_index = 0; depth_index < 4; ++depth_index) {
2999 left_bank_0[depth_index][z] =
3000 left_bank_0[depth_index][z + 1];
3001 left_bank_1[depth_index][z] =
3002 left_bank_1[depth_index][z + 1];
[all …]
/external/deqp/external/openglcts/modules/glesext/texture_cube_map_array/
DesextcTextureCubeMapArrayStencilAttachments.cpp608 …for (glw::GLuint depth_index = 0; depth_index < m_cube_map_array_data[test_index].getDepth(); dept… in fillStencilData() local
617 depth_index; in fillStencilData()
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorImagePatch.h355 … const int depth_index = static_cast<int>(Layout) == static_cast<int>(ColMajor) ? 0 : NumDims - 1;
356 const Index depth = index - (index / m_fastOutputDepth) * m_dimensions[depth_index];
411 … const int depth_index = static_cast<int>(Layout) == static_cast<int>(ColMajor) ? 0 : NumDims - 1;
412 const Index depth = index - (index / m_fastOutputDepth) * m_dimensions[depth_index];
DTensorVolumePatch.h396 … const int depth_index = static_cast<int>(Layout) == static_cast<int>(ColMajor) ? 0 : NumDims - 1;
397 const Index depth = index - (index / m_fastOutputDepth) * m_dimensions[depth_index];
484 … const int depth_index = static_cast<int>(Layout) == static_cast<int>(ColMajor) ? 0 : NumDims - 1;
485 const Index depth = index - (index / m_fastOutputDepth) * m_dimensions[depth_index];
/external/tensorflow/tensorflow/core/kernels/
Deigen_volume_patch.h318 const int depth_index = in coeff() local
322 index - (index / m_fastOutputDepth) * m_dimensions[depth_index]; in coeff()
418 const int depth_index = in packet() local
422 index - (index / m_fastOutputDepth) * m_dimensions[depth_index]; in packet()
/external/tensorflow/tensorflow/core/ops/
Darray_ops.cc2796 int32 depth_index = (axis + new_rank) % new_rank; in __anondb9326b23b02() local
2801 TF_RETURN_IF_ERROR(c->Subshape(indices, 0, depth_index, &front)); in __anondb9326b23b02()
2802 TF_RETURN_IF_ERROR(c->Subshape(indices, depth_index, &back)); in __anondb9326b23b02()