Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/python/ops/
Dinit_ops.py55 def __call__(self, shape, dtype=None, partition_info=None): argument
96 def __call__(self, shape, dtype=None, partition_info=None): argument
112 def __call__(self, shape, dtype=None, partition_info=None): argument
211 def __call__(self, shape, dtype=None, partition_info=None, verify_shape=None): argument
249 def __call__(self, shape, dtype=None, partition_info=None): argument
286 def __call__(self, shape, dtype=None, partition_info=None): argument
328 def __call__(self, shape, dtype=None, partition_info=None): argument
380 def __call__(self, shape, dtype=None, partition_info=None): argument
384 if partition_info is not None:
385 scale_shape = partition_info.full_shape
[all …]
Dvariable_scope.py625 partition_info = _PartitionInfo(
660 partition_info=partition_info,
693 partition_info=None, argument
784 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/libmojo/third_party/catapult/devil/devil/android/
Dfastboot_utils.py59 partition_info = _KNOWN_PARTITIONS[partition]
60 image_file = find_file(partition_info['image'])
63 elif not partition_info.get('optional'):
66 partition_info['image'])
/external/tensorflow/tensorflow/python/estimator/
Dwarm_starting_util.py372 partition_info = None
374 partition_info = variable_scope._PartitionInfo(
394 init(shape=v_shape, partition_info=partition_info))
/external/tensorflow/tensorflow/python/training/
Dcheckpoint_ops.py341 def _initializer(shape, dtype=dtypes.float32, partition_info=None): argument
375 if partition_info is not None:
376 offset = partition_info.single_offset(shape)
Dsaver_large_partitioned_variable_test.py45 init = lambda shape, dtype, partition_info: constant_op.constant( argument
Dcheckpoint_ops_test.py78 def _init_val_initializer(shape, dtype=None, partition_info=None): argument
79 del dtype, partition_info # Unused by this unit-testing initializer.
/external/tensorflow/tensorflow/python/kernel_tests/
Dvariable_scope_test.py1320 partition_info = variable_scope._PartitionInfo(
1322 self.assertEqual(4, partition_info.single_offset([1, 3]))
1325 partition_info = variable_scope._PartitionInfo(
1327 self.assertEqual(0, partition_info.single_offset([9, 3]))
1330 partition_info = variable_scope._PartitionInfo(
1334 partition_info.single_slice_dim(None)
1338 partition_info.single_slice_dim([1, 2, 3])
1342 partition_info.single_slice_dim([6, 3])
1346 partition_info.single_slice_dim([1, 1])
1348 partition_info = variable_scope._PartitionInfo(
[all …]
Dpartitioned_variables_test.py306 def _IotaInitializer(shape, dtype=dtypes.float32, partition_info=None): argument
/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.c427 cpi->mb.partition_info = cpi->mb.pi; in pack_inter_mode_mvs()
565 blockmode = cpi->mb.partition_info->bmi[j].mode; in pack_inter_mode_mvs()
566 blockmv = cpi->mb.partition_info->bmi[j].mv; in pack_inter_mode_mvs()
582 } while (++j < cpi->mb.partition_info->count); in pack_inter_mode_mvs()
590 cpi->mb.partition_info++; in pack_inter_mode_mvs()
594 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/contrib/eager/python/
Dcheckpointable_utils.py59 partition_info=None, **kwargs): argument
82 shape_object.as_list(), dtype=dtype, partition_info=partition_info)
/external/tensorflow/tensorflow/python/feature_column/
Dfeature_column_test.py1721 def _embedding_column_initializer(shape, dtype, partition_info): argument
1724 del partition_info # unused
1765 def _embedding_column_initializer(shape, dtype, partition_info): argument
1768 del partition_info # unused
2044 def _initializer(shape, dtype, partition_info): argument
2045 del shape, dtype, partition_info
2089 def _initializer(shape, dtype, partition_info): argument
2090 del shape, dtype, partition_info
2139 def _initializer(shape, dtype, partition_info): argument
2140 del shape, dtype, partition_info
[all …]
/external/tensorflow/tensorflow/contrib/distributions/python/ops/bijectors/
Dmasked_autoregressive.py374 def masked_initializer(shape, dtype=None, partition_info=None): argument
375 return mask * kernel_initializer(shape, dtype, partition_info)
/external/tensorflow/tensorflow/contrib/layers/python/layers/
Dinitializers.py117 def _initializer(shape, dtype=dtype, partition_info=None): argument
Dlayers.py1891 def beta_initializer(shape, dtype=None, partition_info=None): argument
1892 del partition_info # unused
1895 def gamma_initializer(shape, dtype=None, partition_info=None): argument
1896 del partition_info # unused
Drev_block_lib_test.py290 kernel_initializer=lambda shape, dtype, partition_info: w
/external/tensorflow/tensorflow/contrib/kfac/python/ops/
Dfisher_factors.py99 def inverse_initializer(shape, dtype, partition_info=None): # pylint: disable=unused-argument argument
103 def covariance_initializer(shape, dtype, partition_info=None): # pylint: disable=unused-argument argument
109 def diagonal_covariance_initializer(shape, dtype, partition_info): # pylint: disable=unused-argume… argument
/external/tensorflow/tensorflow/contrib/rnn/python/ops/
Drnn_cell.py1885 def _initializer(shape, dtype=dtype, partition_info=None): argument
1886 del partition_info # Unused.

12