Searched refs:padding_mask (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | partial_batch_padding_handler.py | 35 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)
|
| D | training_distributed.py | 608 padding_handler.padding_mask = dataset.reduce(padding_handler.padding_mask,
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | dense_attention.py | 96 padding_mask = array_ops.expand_dims(
99 scores -= 1.e9 * math_ops.cast(padding_mask, dtype=K.floatx())
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_estimator.py | 3558 padding_mask, features, labels = (
3570 padding_mask = None
3574 padding_mask=padding_mask).as_dict()
3587 def __init__(self, stop, batch_size, padding_mask=None): argument
3590 self._padding_mask = padding_mask
3662 padding_mask = _PaddingSignals._padding_mask(real_batch_size, missing_count,
3665 return padding_mask, features, labels
3671 padding_mask = signals['padding_mask']
3672 batch_size = array_ops.shape(padding_mask)[0]
3682 real_batch_size = batch_size - math_ops.reduce_sum(padding_mask)
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | depthwiseconv_uint8_3x3_filter.h | 4443 int8x8_t padding_mask;
4448 padding_mask = vdup_n_s8(-1);
4453 padding_mask = vshl_u64(padding_mask, vdup_n_s64(8 * copy_remaining));
4508 vbsl_s8(padding_mask, vget_low_s8(padding_reg), half_work_reg);
4524 padding_mask = vshl_u64(padding_mask, vdup_n_s64(8 * copy_remaining));
4579 vbsl_s8(padding_mask, vget_low_s8(padding_reg), half_work_reg);
4607 padding_mask = vcreate_u64(~0xffffff00L);
4622 vbsl_s8(padding_mask, vget_low_s8(padding_reg), half_work_reg);
4644 padding_mask = vshl_u64(padding_mask, vdup_n_s64(8 * copy_remaining));
4646 padding_mask = vset_lane_u8(255, padding_mask, 0);
[all …]
|
| D | depthwiseconv_uint8_transitional.h | 1734 int8x8_t padding_mask;
1739 padding_mask = vdup_n_s8(-1);
1744 padding_mask = vshl_u64(padding_mask, vdup_n_s64(8 * copy_remaining));
1799 vbsl_s8(padding_mask, vget_low_s8(padding_reg), half_work_reg);
1815 padding_mask = vshl_u64(padding_mask, vdup_n_s64(8 * copy_remaining));
1870 vbsl_s8(padding_mask, vget_low_s8(padding_reg), half_work_reg);
1898 padding_mask = vcreate_u64(~0xffffff00L);
1913 vbsl_s8(padding_mask, vget_low_s8(padding_reg), half_work_reg);
1935 padding_mask = vshl_u64(padding_mask, vdup_n_s64(8 * copy_remaining));
1937 padding_mask = vset_lane_u8(255, padding_mask, 0);
[all …]
|