Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/util/
Dpadding.cc56 const int32 depth_index = GetTensorFeatureDimIndex(num_dims, data_format); in CheckValidPadding() local
59 explicit_paddings[2 * depth_index] != 0 || in CheckValidPadding()
60 explicit_paddings[2 * depth_index + 1] != 0) { in CheckValidPadding()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Ddepthwiseconv_uint8_transitional.h2281 for (int depth_index = 0; depth_index < 4; ++depth_index) {
2282 memcpy(selected_data[k_height][depth_index],
2283 &left_bank[k_height][depth_index][offset], 4 - offset);
2284 memcpy(&selected_data[k_height][depth_index][4 - offset],
2285 right_bank[k_height][depth_index], offset);
2716 for (int depth_index = 0; depth_index < 4; ++depth_index) {
2718 left_bank_0[depth_index][z] =
2719 left_bank_0[depth_index][z + 1];
2720 left_bank_1[depth_index][z] =
2721 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.cc2700 int32 depth_index = (axis + new_rank) % new_rank; in __anon7c94107b3c02() local
2705 TF_RETURN_IF_ERROR(c->Subshape(indices, 0, depth_index, &front)); in __anon7c94107b3c02()
2706 TF_RETURN_IF_ERROR(c->Subshape(indices, depth_index, &back)); in __anon7c94107b3c02()