Home
last modified time | relevance | path

Searched refs:padding_mask (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/python/keras/engine/
Dpartial_batch_padding_handler.py35 self.padding_mask = array_ops.zeros(0)
56 def update_mask(self, padding_mask, dataset_batch): argument
62 return K.concatenate([padding_mask, mask], axis=0)
91 padding_mask = K.get_value(self.padding_mask)
92 assert len(padding_mask.shape) == 1
97 padding_mask[:len(prediction_result)]),
108 padding_mask[:len(prediction)]), axis=0)
Dtraining_distributed_v1.py456 padding_handler.padding_mask = dataset.reduce(padding_handler.padding_mask,
/external/tensorflow/tensorflow/python/keras/layers/
Ddense_attention.py127 padding_mask = math_ops.logical_not(scores_mask)
131 scores -= 65504. * math_ops.cast(padding_mask, dtype=scores.dtype)
133 scores -= 1.e9 * math_ops.cast(padding_mask, dtype=scores.dtype)
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Ddepthwiseconv_uint8_3x3_filter.h6817 int8x8_t padding_mask;
6822 padding_mask = vdup_n_s8(-1);
6827 padding_mask = vreinterpret_s8_s64(vshl_s64(
6828 vreinterpret_s64_s8(padding_mask), vdup_n_s64(8 * copy_remaining)));
6893 half_work_reg = vbsl_s8(vreinterpret_u8_s8(padding_mask),
6918 padding_mask = vreinterpret_s8_s64(vshl_s64(
6919 vreinterpret_s64_s8(padding_mask), vdup_n_s64(8 * copy_remaining)));
6977 half_work_reg = vbsl_s8(vreinterpret_u8_s8(padding_mask),
7056 padding_mask = vreinterpret_s8_s64(vshl_s64(
7057 vreinterpret_s64_s8(padding_mask), vdup_n_s64(8 * copy_remaining)));
[all …]
Ddepthwiseconv_uint8_transitional.h1980 int8x8_t padding_mask;
1985 padding_mask = vdup_n_s8(-1);
1990 padding_mask = vreinterpret_s8_s64(vshl_s64(
1991 vreinterpret_s64_s8(padding_mask), vdup_n_s64(8 * copy_remaining)));
2052 half_work_reg = vbsl_s8(vreinterpret_u8_s8(padding_mask),
2071 padding_mask = vreinterpret_s8_s64(vshl_s64(
2072 vreinterpret_s64_s8(padding_mask), vdup_n_s64(8 * copy_remaining)));
2127 half_work_reg = vbsl_s8(vreinterpret_u8_s8(padding_mask),
2194 padding_mask = vreinterpret_s8_s64(vshl_s64(
2195 vreinterpret_s64_s8(padding_mask), vdup_n_s64(8 * copy_remaining)));
[all …]