Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/python/ops/
Dinit_ops.py58 def __call__(self, shape, dtype=None, partition_info=None): argument
112 def __call__(self, shape, dtype=None, partition_info=None): argument
133 def __call__(self, shape, dtype=None, partition_info=None): argument
241 def __call__(self, shape, dtype=None, partition_info=None, verify_shape=None): argument
284 def __call__(self, shape, dtype=None, partition_info=None): argument
326 def __call__(self, shape, dtype=None, partition_info=None): argument
374 def __call__(self, shape, dtype=None, partition_info=None): argument
436 def __call__(self, shape, dtype=None, partition_info=None): argument
440 if partition_info is not None:
441 scale_shape = partition_info.full_shape
[all …]
Dinit_ops_v2.py412 partition_info = None # Keeps logic so can be readded later if necessary
416 if partition_info is not None:
417 scale_shape = partition_info.full_shape
538 partition_info = None # Keeps logic so can be readded later if necessary
540 full_shape = shape if partition_info is None else partition_info.full_shape
545 if partition_info is not None:
546 initializer = array_ops.slice(initializer, partition_info.var_offset,
Dvariable_scope.py742 partition_info = _PartitionInfo(
773 partition_info=partition_info,
806 partition_info=None, argument
906 shape.as_list(), dtype=dtype, partition_info=partition_info)
/external/tensorflow/tensorflow/contrib/framework/python/ops/
Dcheckpoint_ops.py176 def _initializer(shape, dtype=dtypes.float32, partition_info=None): argument
177 del partition_info # Unused by this override.
178 return initializer_fn(shape, dtype, partition_info=primary_partition_info)
Dcheckpoint_ops_test.py64 def _init_val_initializer(shape, dtype=None, partition_info=None): argument
65 del dtype, partition_info # Unused by this unit-testing initializer.
123 def _init_val_initializer(shape, dtype=None, partition_info=None): argument
124 del dtype, partition_info # Unused by this unit-testing initializer.
/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/training/
Dwarm_starting_util.py255 partition_info = None
257 partition_info = variable_scope._PartitionInfo(
302 init(shape=v_shape, partition_info=partition_info))
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)
Dsaver_large_partitioned_variable_test.py46 init = lambda shape, dtype, partition_info: constant_op.constant( argument
Dcheckpoint_ops_test.py80 def _init_val_initializer(shape, dtype=None, partition_info=None): argument
81 del dtype, partition_info # Unused by this unit-testing initializer.
/external/tensorflow/tensorflow/python/tpu/
Dfeature_column_test.py102 def _initializer(shape, dtype, partition_info): argument
105 self.assertIsNone(partition_info)
243 def _initializer(shape, dtype, partition_info): argument
246 self.assertIsNone(partition_info)
/external/tensorflow/tensorflow/contrib/feature_column/python/feature_column/
Dsequence_feature_column_test.py104 def _initializer(shape, dtype, partition_info): argument
107 self.assertIsNone(partition_info)
190 def _initializer(shape, dtype, partition_info): argument
193 self.assertIsNone(partition_info)
914 def _initializer(shape, dtype, partition_info): argument
917 self.assertIsNone(partition_info)
1009 def _initializer(shape, dtype, partition_info): argument
1012 self.assertIsNone(partition_info)
/external/tensorflow/tensorflow/python/feature_column/
Dsequence_feature_column_test.py107 def _initializer(shape, dtype, partition_info): argument
110 self.assertIsNone(partition_info)
194 def _initializer(shape, dtype, partition_info): argument
197 self.assertIsNone(partition_info)
966 def _initializer(shape, dtype, partition_info): argument
969 self.assertIsNone(partition_info)
1061 def _initializer(shape, dtype, partition_info): argument
1064 self.assertIsNone(partition_info)
Dfeature_column_v2_test.py3196 def _embedding_column_initializer(shape, dtype, partition_info): argument
3199 del partition_info # unused
3241 def _embedding_column_initializer(shape, dtype, partition_info): argument
3244 del partition_info # unused
3622 def _initializer(shape, dtype, partition_info): argument
3623 del shape, dtype, partition_info
3667 def _initializer(shape, dtype, partition_info): argument
3668 del shape, dtype, partition_info
3724 def _initializer(shape, dtype, partition_info): argument
3725 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
4937 def _initializer(shape, dtype, partition_info): argument
4940 self.assertIsNone(partition_info)
[all …]
/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()
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()
Dblock.h74 PARTITION_INFO *partition_info; /* work pointer */ member
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.py1733 partition_info = variable_scope._PartitionInfo(
1735 self.assertEqual(4, partition_info.single_offset([1, 3]))
1738 partition_info = variable_scope._PartitionInfo(
1740 self.assertEqual(0, partition_info.single_offset([9, 3]))
1745 partition_info = variable_scope._PartitionInfo(
1749 partition_info.single_slice_dim(None)
1753 partition_info.single_slice_dim([1, 2, 3])
1757 partition_info.single_slice_dim([6, 3])
1761 partition_info.single_slice_dim([1, 1])
1763 partition_info = variable_scope._PartitionInfo(
[all …]
/external/tensorflow/tensorflow/contrib/layers/python/layers/
Dinitializers.py117 def _initializer(shape, dtype=dtype, partition_info=None): argument
/external/tensorflow/tensorflow/contrib/distributions/python/ops/bijectors/
Dmasked_autoregressive.py423 def masked_initializer(shape, dtype=None, partition_info=None): argument
424 return mask * kernel_initializer(shape, dtype, partition_info)
/external/tensorflow/tensorflow/python/training/tracking/
Dutil.py258 partition_info=None, **kwargs): argument
281 shape_object.as_list(), dtype=dtype, partition_info=partition_info)
/external/tensorflow/tensorflow/contrib/rnn/python/kernel_tests/
Dlstm_ops_test.py81 def __call__(self, shape, dtype=dtypes.float16, partition_info=None): argument
85 shape, dtype, partition_info)

12