Home
last modified time | relevance | path

Searched refs:partition_info (Results 1 – 25 of 28) sorted by relevance

12

/external/tensorflow/tensorflow/python/ops/
Dinit_ops.py58 def __call__(self, shape, dtype=None, partition_info=None): argument
111 def __call__(self, shape, dtype=None, partition_info=None): argument
131 def __call__(self, shape, dtype=None, partition_info=None): argument
224 def __call__(self, shape, dtype=None, partition_info=None, verify_shape=None): argument
265 def __call__(self, shape, dtype=None, partition_info=None): argument
305 def __call__(self, shape, dtype=None, partition_info=None): argument
351 def __call__(self, shape, dtype=None, partition_info=None): argument
411 def __call__(self, shape, dtype=None, partition_info=None): argument
415 if partition_info is not None:
416 scale_shape = partition_info.full_shape
[all …]
Dinit_ops_v2.py536 partition_info = None # Keeps logic so can be readded later if necessary
540 if partition_info is not None:
541 scale_shape = partition_info.full_shape
698 partition_info = None # Keeps logic so can be readded later if necessary
700 full_shape = shape if partition_info is None else partition_info.full_shape
705 if partition_info is not None:
706 initializer = array_ops.slice(initializer, partition_info.var_offset,
Dvariable_scope.py736 partition_info = _PartitionInfo(
768 partition_info=partition_info,
803 partition_info=None, argument
908 partition_info=partition_info)
/external/swiftshader/third_party/astc-encoder/Source/
Dastc_partition_tables.cpp79 partition_info* pi in partition_table_zap_equal_elements()
229 partition_info* pt in generate_one_partition_table()
284 partition_info *par_tab2 = bsd->partitions; in init_partition_tables()
285 partition_info *par_tab3 = par_tab2 + PARTITION_COUNT; in init_partition_tables()
286 partition_info *par_tab4 = par_tab3 + PARTITION_COUNT; in init_partition_tables()
287 partition_info *par_tab1 = par_tab4 + PARTITION_COUNT; in init_partition_tables()
Dastc_codec_internals.h64 struct partition_info struct
137 partition_info partitions[(3*PARTITION_COUNT)+1];
319 static inline const partition_info *get_partition_table( in get_partition_table()
Dastc_decompress_symbolic.cpp203 const partition_info *pt = get_partition_table(bsd, partition_count); in decompress_symbolic_block()
/external/chromium-trace/catapult/devil/devil/android/
Dfastboot_utils.py61 partition_info = _KNOWN_PARTITIONS[partition]
62 image_file = find_file(partition_info['image'])
65 elif not partition_info.get('optional'):
68 partition_info['image'])
/external/tensorflow/tensorflow/python/feature_column/
Ddense_features_test.py69 def _embedding_column_initializer(shape, dtype, partition_info=None): argument
72 del partition_info # unused
114 def _embedding_column_initializer(shape, dtype, partition_info=None): argument
115 offset = partition_info._var_offset[0]
166 def _embedding_column_initializer(shape, dtype, partition_info=None): argument
169 del partition_info # unused
547 def _initializer(shape, dtype, partition_info=None): argument
548 del shape, dtype, partition_info
604 def _initializer(shape, dtype, partition_info=None): argument
605 del shape, dtype, partition_info
Ddense_features_v2_test.py68 def _embedding_column_initializer(shape, dtype, partition_info=None): argument
71 del partition_info # unused
113 def _embedding_column_initializer(shape, dtype, partition_info=None): argument
116 del partition_info # unused
494 def _initializer(shape, dtype, partition_info=None): argument
495 del shape, dtype, partition_info
551 def _initializer(shape, dtype, partition_info=None): argument
552 del shape, dtype, partition_info
Dsequence_feature_column_test.py118 def _initializer(shape, dtype, partition_info=None): argument
121 self.assertIsNone(partition_info)
204 def _initializer(shape, dtype, partition_info=None): argument
207 self.assertIsNone(partition_info)
1015 def _initializer(shape, dtype, partition_info=None): argument
1018 self.assertIsNone(partition_info)
1110 def _initializer(shape, dtype, partition_info=None): argument
1113 self.assertIsNone(partition_info)
Dfeature_column_v2_test.py3209 def _embedding_column_initializer(shape, dtype, partition_info=None): argument
3212 del partition_info # unused
3254 def _embedding_column_initializer(shape, dtype, partition_info=None): argument
3257 del partition_info # unused
3631 def _initializer(shape, dtype, partition_info=None): argument
3632 del shape, dtype, partition_info
3688 def _initializer(shape, dtype, partition_info=None): argument
3689 del shape, dtype, partition_info
5063 def _initializer(shape, dtype, partition_info=None): argument
5064 del shape, dtype, partition_info
[all …]
Dfeature_column_test.py2654 def _embedding_column_initializer(shape, dtype, partition_info): argument
2657 del partition_info # unused
2699 def _embedding_column_initializer(shape, dtype, partition_info): argument
2702 del partition_info # unused
3143 def _initializer(shape, dtype, partition_info): argument
3144 del shape, dtype, partition_info
3194 def _initializer(shape, dtype, partition_info): argument
3195 del shape, dtype, partition_info
4401 def _initializer(shape, dtype, partition_info=None): argument
4402 del shape, dtype, partition_info
[all …]
/external/tensorflow/tensorflow/python/training/
Dcheckpoint_ops.py342 def _initializer(shape, dtype=dtypes.float32, partition_info=None): argument
376 if partition_info is not None:
377 offset = partition_info.single_offset(shape)
Dwarm_starting_util.py264 partition_info = None
266 partition_info = variable_scope._PartitionInfo(
310 init(shape=v_shape, partition_info=partition_info))
Dsaver_large_partitioned_variable_test.py46 init = lambda shape, dtype, partition_info: constant_op.constant( argument
Dcheckpoint_ops_test.py81 def _init_val_initializer(shape, dtype=None, partition_info=None): argument
82 del dtype, partition_info # Unused by this unit-testing initializer.
/external/tensorflow/tensorflow/python/tpu/
Dfeature_column_v2_test.py109 def _initializer(shape, dtype, partition_info=None): argument
112 self.assertIsNone(partition_info)
296 def _initializer(shape, dtype, partition_info=None): argument
299 self.assertIsNone(partition_info)
488 def _initializer(shape, dtype, partition_info=None): argument
491 self.assertIsNone(partition_info)
Dfeature_column_test.py127 def _initializer(shape, dtype, partition_info): argument
130 self.assertIsNone(partition_info)
268 def _initializer(shape, dtype, partition_info): argument
271 self.assertIsNone(partition_info)
/external/libvpx/libvpx/vp8/encoder/
Drdopt.c870 x->partition_info->bmi[i].mode = m; in labels2mode()
871 x->partition_info->bmi[i].mv.as_int = this_mv->as_int; in labels2mode()
1178 bsi->mvs[i].as_mv = x->partition_info->bmi[i].mv.as_mv; in rd_check_segment()
1179 bsi->modes[i] = x->partition_info->bmi[i].mode; in rd_check_segment()
1319 x->partition_info->count = vp8_mbsplit_count[bsi.segment_num]; in vp8_rd_pick_best_mbsegmentation()
1321 for (i = 0; i < x->partition_info->count; ++i) { in vp8_rd_pick_best_mbsegmentation()
1326 x->partition_info->bmi[i].mode = bsi.modes[j]; in vp8_rd_pick_best_mbsegmentation()
1327 x->partition_info->bmi[i].mv.as_mv = bsi.mvs[j].as_mv; in vp8_rd_pick_best_mbsegmentation()
1332 x->partition_info->bmi[15].mv.as_int = bsi.mvs[15].as_int; in vp8_rd_pick_best_mbsegmentation()
1565 for (i = 0; i < x->partition_info->count; ++i) { in rd_update_mvcount()
[all …]
Dethreading.c280 x->partition_info++; in thread_encoding_proc()
291 x->partition_info++; in thread_encoding_proc()
305 x->partition_info += xd->mode_info_stride * cpi->encoding_thread_count; in thread_encoding_proc()
454 mb->partition_info = x->pi + x->e_mbd.mode_info_stride * (i + 1); in vp8cx_init_mbrthread_data()
Dblock.h74 PARTITION_INFO *partition_info; /* work pointer */ member
Dbitstream.c420 cpi->mb.partition_info = cpi->mb.pi; in pack_inter_mode_mvs()
532 blockmode = cpi->mb.partition_info->bmi[j].mode; in pack_inter_mode_mvs()
533 blockmv = cpi->mb.partition_info->bmi[j].mv; in pack_inter_mode_mvs()
546 } while (++j < cpi->mb.partition_info->count); in pack_inter_mode_mvs()
554 cpi->mb.partition_info++; in pack_inter_mode_mvs()
558 cpi->mb.partition_info++; in pack_inter_mode_mvs()
Dencodeframe.c555 x->partition_info++; in encode_mb_row()
572 x->partition_info++; in encode_mb_row()
588 x->partition_info = x->pi; in init_encode_frame_mb_context()
789 x->partition_info += xd->mode_info_stride * cpi->encoding_thread_count; in vp8_encode_frame()
/external/tensorflow/tensorflow/python/kernel_tests/
Dvariable_scope_test.py1765 partition_info = variable_scope._PartitionInfo(
1767 self.assertEqual(4, partition_info.single_offset([1, 3]))
1770 partition_info = variable_scope._PartitionInfo(
1772 self.assertEqual(0, partition_info.single_offset([9, 3]))
1777 partition_info = variable_scope._PartitionInfo(
1781 partition_info.single_slice_dim(None)
1785 partition_info.single_slice_dim([1, 2, 3])
1789 partition_info.single_slice_dim([6, 3])
1793 partition_info.single_slice_dim([1, 1])
1795 partition_info = variable_scope._PartitionInfo(
[all …]
/external/tensorflow/tensorflow/python/training/tracking/
Dutil.py389 partition_info=None, argument
416 shape_object.as_list(), dtype=dtype, partition_info=partition_info)

12