Home
last modified time | relevance | path

Searched refs:target_dim (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/python/keras/layers/
Dwrappers_test.py544 target_dim = 2 * output_dim if mode == 'concat' else output_dim
545 y = np.random.random((samples, target_dim))
610 target_dim = 2 * output_dim if mode == 'concat' else output_dim
611 y = np.random.random((samples, target_dim))
643 target_dim = 2 * output_dim if mode == 'concat' else output_dim
644 y = np.random.random((samples, target_dim))
1109 target_dim = 2 * output_dim if mode == 'concat' else output_dim
1110 y = np.random.random((samples, target_dim))
Dcudnn_recurrent_test.py411 target_dim = 2 * output_dim if mode == 'concat' else output_dim
412 y = np.random.random((samples, target_dim))
Dcore.py610 target_dim = input_shape[dim]
611 output_shape[i + 1] = target_dim
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.cc1100 int64 target_dim = source_target_dims[0].second; in ReshardWithAllToAll() local
1102 sharding().tile_assignment().dim(target_dim); in ReshardWithAllToAll()
1115 } else if (target_dim == j) { in ReshardWithAllToAll()
1134 sharding().tile_assignment().dim(target_dim); in ReshardWithAllToAll()
1135 temp_target_tile_dims[target_dim] = in ReshardWithAllToAll()
1144 target_dim, in ReshardWithAllToAll()
1145 RoundUpToNearest(padded_shape.dimensions(target_dim), in ReshardWithAllToAll()
1146 temp_target.tile_assignment().dim(target_dim))); in ReshardWithAllToAll()
1156 if (dim == target_dim) { in ReshardWithAllToAll()
1173 if (i == target_dim) { in ReshardWithAllToAll()
[all …]
Dspmd_partitioner_util.cc1884 int64 target_dim = target->tile_assignment().dim(target_dims[i]); in GatherOperandsShardedAcrossParallelDims() local
1886 if (target_dim < to_adjust_dim) { in GatherOperandsShardedAcrossParallelDims()
1889 if (target_dim == to_adjust_dim) { in GatherOperandsShardedAcrossParallelDims()
1892 int64 ratio = target_dim / to_adjust_dim; in GatherOperandsShardedAcrossParallelDims()
1893 if (target_dim % to_adjust_dim != 0 || in GatherOperandsShardedAcrossParallelDims()
/external/tensorflow/tensorflow/lite/g3doc/models/style_transfer/
Doverview.ipynb224 "def preprocess_image(image, target_dim):\n",
228 " scale = target_dim / short_dim\n",
233 " image = tf.image.resize_with_crop_or_pad(image, target_dim, target_dim)\n",
/external/tensorflow/tensorflow/python/keras/engine/
Dtraining_utils_v1.py832 for target_dim, out_dim in zip(y.shape[1:], shape[1:]):
833 if out_dim is not None and target_dim != out_dim:
/external/llvm-project/polly/lib/External/isl/
Disl_transitive_closure.c2560 isl_space *target_dim; in isl_map_transitive_closure() local
2580 target_dim = isl_map_get_space(map); in isl_map_transitive_closure()
2582 map = isl_map_reset_space(map, target_dim); in isl_map_transitive_closure()