Searched refs:input_col (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | local.py | 463 input_row, input_col = input_shape[1:-1]
466 input_row, input_col = input_shape[2:]
468 if input_row is None or input_col is None:
475 output_col = conv_utils.conv_output_length(input_col, self.kernel_size[1],
493 self.kernel_shape = (input_filter, input_row, input_col, self.filters,
496 self.kernel_shape = (input_row, input_col, input_filter,
507 input_shape=(input_row, input_col),
516 input_row * input_col * input_filter)
520 input_shape=(input_row, input_col),
|
| /external/llvm-project/llvm/test/CodeGen/Thumb2/ |
| D | mve-postinc-lsr.ll | 206 define i8* @test(i8* nocapture readonly %input_row, i8* nocapture readonly %input_col, i16 zeroext …
300 %add.ptr9 = getelementptr inbounds i8, i8* %input_col, i32 %conv5
302 %add.ptr12 = getelementptr inbounds i8, i8* %input_col, i32 %mul11
304 %add.ptr15 = getelementptr inbounds i8, i8* %input_col, i32 %mul14
343 %ip_c0.0108 = phi i8* [ %add.ptr27, %for.body24 ], [ %input_col, %for.body24.preheader ]
389 define i8* @test_optsize(i8* nocapture readonly %input_row, i8* nocapture readonly %input_col, i16 …
482 %add.ptr9 = getelementptr inbounds i8, i8* %input_col, i32 %conv5
484 %add.ptr12 = getelementptr inbounds i8, i8* %input_col, i32 %mul11
486 %add.ptr15 = getelementptr inbounds i8, i8* %input_col, i32 %mul14
525 %ip_c0.0108 = phi i8* [ %add.ptr27, %for.body24 ], [ %input_col, %for.body24.preheader ]
[all …]
|
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorReductionCuda.h | 682 const Index input_col = i % num_preserved_coeffs;
687 typename Self::CoeffReturnType val = input.m_impl.coeff(j * num_preserved_coeffs + input_col);
690 atomicReduce(&(output[input_col]), reduced_val, reducer);
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | eigen_spatial_convolutions-inl.h | 935 const Index input_col = m_colIndex + col * m_base_mapper.m_in_col_strides; in padOrSkipCol() local
937 ? input_col in padOrSkipCol()
938 : ((input_col >= 0) in padOrSkipCol()
939 ? (input_col / m_base_mapper.m_fastInputColStride) in padOrSkipCol()
943 (input_col != *orig_col * m_base_mapper.m_patch_col_inflate_strides); in padOrSkipCol()
|