Searched refs:partition_offset (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | parameter_server_strategy_v2.py | 630 partition_offset = (offsets[shard_index],) + (0,) * len(shape[1:])
637 partition_offset=partition_offset)
660 offset=partition_offset))
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | initializers_test.py | 270 shape=(4, 2), partition_shape=(2, 2), partition_offset=(0, 0))
282 shape=(4, 2), partition_shape=(2, 2), partition_offset=(0, 0))
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | init_ops_v2_test.py | 81 partition_offset = (0, 0)
88 partition_offset=partition_offset))
91 got, array_ops.slice(full_value, partition_offset, partition_shape))
|
| /external/libwebm/ |
| D | webm_info.cc | 823 uint32_t partition_offset; in PrintSubSampleEncryption() local
824 memcpy(&partition_offset, data, sizeof(partition_offset)); in PrintSubSampleEncryption()
825 partition_offset = libwebm::bigendian_to_host(partition_offset); in PrintSubSampleEncryption()
826 fprintf(o, " off[%d]:%u", i, partition_offset); in PrintSubSampleEncryption()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner.cc | 2400 auto partition_offset = add_hlo(HloInstruction::CreateBinary( in HandleDynamicUpdateSlice() local
2405 compare_shape, new_indices[dim], partition_offset, in HandleDynamicUpdateSlice()
2434 partition_offset)), in HandleDynamicUpdateSlice()
|