/external/tensorflow/tensorflow/contrib/tensor_forest/hybrid/python/ops/ |
D | training_ops.py | 73 dl_du = array_ops.expand_dims(grad, 2) 81 du_df = array_ops.expand_dims( 95 df_dx = -array_ops.expand_dims(tree_weights_tensor, 0) 102 df_dt = -array_ops.expand_dims(input_data_tensor, 1) 108 df_db = array_ops.expand_dims( 109 array_ops.expand_dims(array_ops.ones_like(tree_thresholds_tensor), 0), 2) 159 dl_du = array_ops.expand_dims(unpack_path_op(path_tensor, routing_grad), 2) 167 du_df = array_ops.expand_dims(du_df_raw, 2) 175 df_dx = array_ops.expand_dims(df_dx_raw, 0) 188 df_db = array_ops.expand_dims(array_ops.expand_dims(df_db_raw, 0), 2) [all …]
|
/external/tensorflow/tensorflow/contrib/crf/python/kernel_tests/ |
D | crf_test.py | 66 inputs=array_ops.expand_dims(inputs, 0), 67 tag_indices=array_ops.expand_dims(tag_indices, 0), 68 sequence_lengths=array_ops.expand_dims(sequence_lengths, 0), 101 inputs=array_ops.expand_dims(inputs, 0), 102 tag_bitmap=array_ops.expand_dims(tag_bitmap, 0), 103 sequence_lengths=array_ops.expand_dims(sequence_lengths, 0), 129 tag_indices=array_ops.expand_dims(tag_indices, 0), 130 sequence_lengths=array_ops.expand_dims(sequence_lengths, 0), 131 inputs=array_ops.expand_dims(inputs, 0)) 145 tag_indices=array_ops.expand_dims(tag_indices, 0), [all …]
|
/external/tensorflow/tensorflow/core/api_def/base_api/ |
D | api_def_ExpandDims.pbtxt | 28 channels]`, you can make it a batch of 1 image with `expand_dims(image, 0)`, 35 shape(expand_dims(t, 0)) ==> [1, 2] 36 shape(expand_dims(t, 1)) ==> [2, 1] 37 shape(expand_dims(t, -1)) ==> [2, 1] 40 shape(expand_dims(t2, 0)) ==> [1, 2, 3, 5] 41 shape(expand_dims(t2, 2)) ==> [2, 3, 1, 5] 42 shape(expand_dims(t2, 3)) ==> [2, 3, 5, 1]
|
/external/tensorflow/tensorflow/contrib/factorization/python/ops/ |
D | factorization_ops_test_utils.py | 86 ind = (np.concatenate((np.expand_dims(indices[1], 1), 87 np.expand_dims(indices[0], 1)), 1).astype(np.int64) if 88 transpose else np.concatenate((np.expand_dims(indices[0], 1), 89 np.expand_dims(indices[1], 1)), 118 wr = (array_ops.expand_dims(row_weights, 1) if row_weights is not None 120 wc = (array_ops.expand_dims(col_weights, 0) if col_weights is not None
|
D | gmm_ops.py | 182 means = array_ops.expand_dims(initial_means, 1) 185 means = array_ops.expand_dims( 193 array_ops.expand_dims(cov, 0), [self._num_classes, 1, 1]) 198 array_ops.expand_dims(array_ops.diag_part(cov), 0), 273 shard = array_ops.expand_dims(shard, 0) 319 cov_expanded = array_ops.expand_dims(1.0 / (self._covs + 1e-3), 2) 359 probs = array_ops.expand_dims(self._probs[shard_id], 0) 381 w_mul_x = array_ops.expand_dims( 390 array_ops.expand_dims(x_trans[k, :, :] * self._w[shard_id][:, k], 0) 433 new_covs.append(array_ops.expand_dims(new_cov, 0)) [all …]
|
/external/tensorflow/tensorflow/contrib/image/python/kernel_tests/ |
D | sparse_image_warp_test.py | 95 np.float32(np.expand_dims(control_point_locations, 0))) 137 np.float32(np.expand_dims(control_point_locations, 0)), 142 np.float32(np.expand_dims(control_point_displacements, 0)), 188 np.expand_dims(np.float32(control_points[:, [1, 0]]), 0)) 190 np.expand_dims(np.float32(control_point_displacements[:, [1, 0]]), 0)) 191 float_image = np.expand_dims(np.float32(input_image) / 255, 0) 230 np.float32(np.expand_dims(control_point_locations, 0))) 233 np.float32(np.expand_dims(control_point_displacements, 0)))
|
/external/tensorflow/tensorflow/tools/compatibility/testdata/ |
D | test_file_v0_11.py | 98 tf.expand_dims([[1, 2], [3, 4]], axis=1).eval(), 124 self.assertAllEqual(tf.expand_dims(tf.squeeze(a, [0]), 0).eval(), 126 self.assertAllEqual(tf.squeeze(tf.expand_dims(a, 1), [1]).eval(), 129 tf.expand_dims(tf.squeeze([[1, 2, 3]], axis=[0]), dim=0).eval(), a) 131 tf.squeeze(tf.expand_dims([[1, 2, 3]], dim=1), axis=[1]).eval(), a) 134 tf.squeeze(tf.expand_dims([[1, 2, 3]], dim=1), axis=[1]).eval(), a) 165 batched_mat = tf.expand_dims(mat, [0]) 168 self.assertAllEqual(result_batched, np.expand_dims(result, 0))
|
/external/tensorflow/tensorflow/contrib/crf/python/ops/ |
D | crf.py | 281 offsets = array_ops.expand_dims( 283 offsets += array_ops.expand_dims(math_ops.range(max_seq_len) * num_tags, 0) 351 self._transition_params = array_ops.expand_dims(transition_params, 0) 375 state = array_ops.expand_dims(state, 2) 411 v = np.expand_dims(trellis[t - 1], 1) + transition_params 437 self._transition_params = array_ops.expand_dims(transition_params, 0) 463 state = array_ops.expand_dims(state, 2) # [B, O, 1] 512 new_tags = array_ops.expand_dims( 540 decode_tags = array_ops.expand_dims( 575 initial_state = array_ops.expand_dims(initial_state, axis=-1) # [B, 1]
|
/external/tensorflow/tensorflow/python/ops/ |
D | ctc_ops.py | 404 array_ops.expand_dims(indices, 0), [batch_size, 1, 1]) 405 batch_idx = array_ops.expand_dims(math_ops.range(batch_size), 1) * [1, 0, 0] 406 indices += array_ops.expand_dims(batch_idx, 1) 412 return array_ops.expand_dims(trans, 0) + label_to_label 462 one_hot = array_ops.expand_dims(one_hot, axis=0) 463 ilabel_log_probs = array_ops.expand_dims(ilabel_log_probs, axis=2) 480 one_hot = array_ops.expand_dims(one_hot, axis=0) 481 label_states = array_ops.expand_dims(label_states, axis=3) 506 indices = unique_y + array_ops.expand_dims(batch_offset, axis=-1) 837 indices=array_ops.expand_dims( [all …]
|
D | init_ops.py | 684 array_ops.expand_dims(q, 0), shape) 687 array_ops.expand_dims(q, 0), shape) 691 array_ops.expand_dims(q, 0), shape) 890 return array_ops.expand_dims(array_ops.expand_dims(orth, 0), 0) 1015 return array_ops.expand_dims(orth, 0) 1164 return array_ops.expand_dims( 1165 array_ops.expand_dims( 1166 array_ops.expand_dims(orth, 0), 0), 0)
|
/external/tensorflow/tensorflow/python/kernel_tests/ |
D | conv1d_test.py | 36 x = array_ops.expand_dims(x, 0) # Add batch dimension 37 x = array_ops.expand_dims(x, 2) # And depth dimension 39 filters = array_ops.expand_dims(filters, 1) # in_channels 40 filters = array_ops.expand_dims(filters, 2) # out_channels
|
D | attention_ops_test.py | 60 array_ops.expand_dims(array_ops.expand_dims(t_rows, 0), 3), 74 array_ops.expand_dims(array_ops.expand_dims(t_cols, 0), 3),
|
D | shape_ops_test.py | 178 np_ans = np.expand_dims(x, axis=dim) 180 tensor = array_ops.expand_dims(x, dim) 235 self.assertRaises(ValueError, array_ops.expand_dims, 237 self.assertRaises(ValueError, array_ops.expand_dims, 239 self.assertRaises(ValueError, array_ops.expand_dims, 241 self.assertRaises(ValueError, array_ops.expand_dims, 249 squeezed = array_ops.expand_dims(inp, 1) 259 self.assertAllEqual([7], array_ops.expand_dims(inp, 0).eval()) 260 self.assertAllEqual([7], array_ops.expand_dims(inp, -1).eval()) 263 self.assertAllEqual([True], array_ops.expand_dims(inp, 0).eval()) [all …]
|
D | draw_bounding_box_op_test.py | 84 bboxes = array_ops.expand_dims(bboxes, 0) 87 image = array_ops.expand_dims(image, 0)
|
/external/tensorflow/tensorflow/contrib/solvers/python/ops/ |
D | lanczos.py | 97 return array_ops.expand_dims(tarray.read(i), -1) 226 alpha = array_ops.expand_dims(alpha, 0) 228 beta = array_ops.expand_dims(beta, 0) 231 beta = array_ops.expand_dims(beta[:-1], 0) 233 zero_column = array_ops.expand_dims(
|
D | linear_equations.py | 107 rhs = array_ops.expand_dims(rhs, -1) 109 x = array_ops.expand_dims( 113 x = array_ops.expand_dims(x, -1)
|
D | least_squares.py | 97 rhs = array_ops.expand_dims(rhs, -1) 101 x = array_ops.expand_dims(
|
/external/tensorflow/tensorflow/python/keras/utils/ |
D | kernelized_utils.py | 32 return array_ops.expand_dims(u, 0) 48 array_ops.expand_dims(x_matrix, 1), [1, y_shape[0], 1]) 50 array_ops.expand_dims(y_matrix, 0), [x_shape[0], 1, 1])
|
/external/tensorflow/tensorflow/contrib/image/python/ops/ |
D | interpolate_spline.py | 45 x_norm_squared_tile = array_ops.expand_dims(x_norm_squared, 2) 46 y_norm_squared_tile = array_ops.expand_dims(y_norm_squared, 1) 71 x_norm_squared_tile = array_ops.expand_dims(x_norm_squared, 2) 126 batch_identity_matrix = array_ops.expand_dims(
|
/external/tensorflow/tensorflow/contrib/seq2seq/python/ops/ |
D | beam_search_decoder.py | 91 tiled = array_ops.tile(array_ops.expand_dims(t, 1), tiling) 148 beam_ids = array_ops.expand_dims( 149 array_ops.expand_dims(math_ops.range(beam_width), 0), 0) 698 array_ops.expand_dims(self._start_tokens, 1), [1, self._beam_width]) 882 array_ops.expand_dims(self._start_tokens, 1), [1, self._beam_width]) 978 total_probs = array_ops.expand_dims(beam_state.log_probs, 2) + step_log_probs 989 lengths_to_add *= array_ops.expand_dims(add_mask, 2) 991 lengths_to_add + array_ops.expand_dims(prediction_lengths, 2)) 999 attention_probs *= array_ops.expand_dims( 1150 array_ops.expand_dims(prob, -1) for prob in probs_per_attn_layer] [all …]
|
/external/tensorflow/tensorflow/lite/kernels/ |
D | expand_dims.cc | 27 namespace expand_dims { namespace 107 static TfLiteRegistration r = {nullptr, nullptr, expand_dims::Prepare, in Register_EXPAND_DIMS() 108 expand_dims::Eval}; in Register_EXPAND_DIMS()
|
/external/tensorflow/tensorflow/contrib/training/python/training/ |
D | sequence_queueing_state_saver.py | 1079 sequence_count_vec = array_ops.expand_dims(sequence_count, 0) 1099 array_ops.slice(current_keys, [1], [-1]), array_ops.expand_dims( 1114 array_ops.expand_dims(sequence_count, 0), 1115 array_ops.expand_dims(self._num_unroll, 0), 1131 array_ops.expand_dims(v, 0), 1134 array_ops.expand_dims(sequence_count, 0), 1675 values = array_ops.tile(sp_tensor.values, array_ops.expand_dims(n, 0)) 1677 [array_ops.expand_dims(value_length, 0), sp_tensor.dense_shape], 0) 1682 array_ops.expand_dims(sp_tensor.indices, 0), 1697 ind = array_ops.expand_dims(ind, 1) [all …]
|
/external/tensorflow/tensorflow/examples/wav_to_spectrogram/ |
D | wav_to_spectrogram.cc | 69 Output expand_dims = in WavToSpectrogram() local 71 Output squeeze = Squeeze(root.WithOpName("squeeze"), expand_dims, in WavToSpectrogram()
|
/external/tensorflow/tensorflow/lite/testing/nnapi_tflite_zip_tests/ |
D | not_supported.txt | 133 expand_dims/expand_dims_axis_value=0,constant_axis=True,input_shape=[5,4],input_type=tf.float32 134 expand_dims/expand_dims_axis_value=1,constant_axis=True,input_shape=[5,4],input_type=tf.float32 135 expand_dims/expand_dims_axis_value=2,constant_axis=True,input_shape=[5,4],input_type=tf.float32 136 expand_dims/expand_dims_axis_value=-1,constant_axis=True,input_shape=[5,4],input_type=tf.float32 137 expand_dims/expand_dims_axis_value=-2,constant_axis=True,input_shape=[5,4],input_type=tf.float32 138 expand_dims/expand_dims_axis_value=-3,constant_axis=True,input_shape=[5,4],input_type=tf.float32 139 expand_dims/expand_dims_axis_value=0,constant_axis=False,input_shape=[5,4],input_type=tf.float32 140 expand_dims/expand_dims_axis_value=1,constant_axis=False,input_shape=[5,4],input_type=tf.float32 141 expand_dims/expand_dims_axis_value=2,constant_axis=False,input_shape=[5,4],input_type=tf.float32 142 expand_dims/expand_dims_axis_value=-1,constant_axis=False,input_shape=[5,4],input_type=tf.float32 [all …]
|
/external/tensorflow/tensorflow/contrib/constrained_optimization/python/ |
D | swap_regret_optimizer.py | 106 matrix, standard_ops.expand_dims(eigenvector, 1))[:, 0] 366 (standard_ops.expand_dims(objective, 0), proxy_constraints), axis=0) 373 standard_ops.expand_dims( 375 standard_ops.expand_dims(distribution, 0))
|