Searched refs:set_shape (Results 1 – 25 of 131) sorted by relevance
123456
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | variable_ops_test.py | 68 p = state_ops.variable_op([1, 2], dtypes.float32, set_shape=False)
87 var = state_ops.variable_op(value.shape, dtypes.float32, set_shape=False)
94 var = state_ops.variable_op(value.shape, dtypes.float32, set_shape=False)
123 var = state_ops.variable_op([1, 2], dtypes.float32, set_shape=False)
131 var = state_ops.variable_op([1, 2], dtypes.float32, set_shape=False)
146 var = state_ops.variable_op([1, 2], dtypes.float32, set_shape=False)
160 var = state_ops.variable_op([1, 2], dtypes.float32, set_shape=False)
|
| D | tensordot_op_test.py | 113 a.set_shape([None, 2])
114 b.set_shape([2, 3])
123 a.set_shape([2, 2])
124 b.set_shape([2, None])
|
| D | neon_depthwise_conv_op_test.py | 120 t1.set_shape(tensor_in_sizes)
210 t1.set_shape(tensor_in_sizes)
|
| /external/tensorflow/tensorflow/contrib/model_pruning/examples/cifar10/ |
| D | cifar10_input.py | 187 float_image.set_shape([height, width, 3])
188 read_input.label.set_shape([1])
250 float_image.set_shape([height, width, 3])
251 read_input.label.set_shape([1])
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | state_ops.py | 97 def variable_op(shape, dtype, name="Variable", set_shape=True, container="", argument
100 if not set_shape:
106 if set_shape:
107 ret.set_shape(shape)
|
| D | string_ops.py | 100 indices.set_shape([None, 2])
101 values.set_shape([None])
102 shape.set_shape([2])
|
| D | image_ops_impl.py | 241 result.set_shape([None, None, None])
243 result.set_shape(image_shape)
429 result.set_shape([None, None, image.get_shape()[2]])
461 result.set_shape([shape[0], None, None, shape[3]])
544 image.set_shape([None, None, depth])
590 image.set_shape([None] * 4)
623 padded.set_shape(padded_shape)
673 image.set_shape([None] * 4)
703 cropped.set_shape(cropped_shape)
750 image.set_shape([None] * 4)
[all …]
|
| D | rnn.py | 75 x_t.set_shape(
272 output.set_shape(flat_output.get_shape())
275 substate.set_shape(flat_substate.get_shape())
309 input_.set_shape(input_shape)
319 r.set_shape(input_shape)
784 input_.set_shape(shape[1:])
841 output.set_shape(shape)
1299 output.set_shape(tensor_shape.TensorShape(shape))
|
| D | sparse_ops.py | 1547 output_indices.set_shape([None, rank])
1548 output_shape.set_shape([rank])
1617 output_indices.set_shape([None, rank])
1618 output_shape.set_shape([rank])
2178 output_indices.set_shape([None, rank])
2179 output_shape.set_shape([rank])
|
| D | data_flow_ops.py | 450 output.set_shape(shape)
492 output.set_shape(
534 output.set_shape(tensor_shape.TensorShape([None]).concatenate(shape))
1093 op.outputs[0].set_shape(tensor_shape.vector(batch_dim)) # indices
1094 op.outputs[1].set_shape(tensor_shape.vector(batch_dim)) # keys
1096 output.set_shape(
1315 out.set_shape(self._shape)
1662 output.set_shape(self._shapes[i])
|
| D | sparse_grad.py | 95 a_val_grad.set_shape(op.inputs[1].get_shape())
96 b_val_grad.set_shape(op.inputs[4].get_shape())
|
| D | control_flow_ops.py | 271 result.set_shape(data.get_shape())
290 indices.set_shape(data.indices.get_shape())
301 dense_shape.set_shape(data.dense_shape.get_shape())
311 dense_shape.set_shape(data.dense_shape.get_shape())
577 var.set_shape(shape)
589 var.values.set_shape(shape)
590 var.indices.set_shape(tensor_shape.TensorShape([shape[0]]))
592 var.dense_shape.set_shape(tensor_shape.TensorShape([shape.ndims]))
601 var.values.set_shape(tensor_shape.TensorShape([None]))
602 var.indices.set_shape(tensor_shape.TensorShape([None, shape.ndims]))
[all …]
|
| /external/tensorflow/tensorflow/contrib/seq2seq/python/ops/ |
| D | beam_search_decoder.py | 90 tiled.set_shape(
375 reshaped_t.set_shape(
417 reshaped_t.set_shape(expected_reshaped_shape)
581 next_beam_scores.set_shape([static_batch_size, beam_width])
582 word_indices.set_shape([static_batch_size, beam_width])
694 penalty_factor.set_shape(()) # penalty should be a scalar.
814 output.set_shape(final_static_shape)
|
| /external/tensorflow/tensorflow/contrib/data/python/ops/ |
| D | grouping.py | 133 key.set_shape([])
156 arg.set_shape(shape)
182 key.set_shape([])
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_SetSize.pbtxt | 16 name: "set_shape"
32 and `set_shape`. The last dimension contains values in a set, duplicates are
|
| /external/tensorflow/tensorflow/contrib/lookup/ |
| D | lookup_ops.py | 413 values.set_shape(keys.get_shape().concatenate(self._value_shape))
461 exported_values.set_shape(exported_keys.get_shape().concatenate(
615 values.set_shape(
665 exported_values.set_shape(exported_keys.get_shape().concatenate(
|
| /external/tensorflow/tensorflow/contrib/slim/python/slim/data/ |
| D | prefetch_queue_test.py | 169 p1.set_shape(var_shape)
171 p2.set_shape(var_shape)
173 p3.set_shape(var_shape)
|
| /external/tensorflow/tensorflow/python/ops/distributions/ |
| D | distribution.py | 1162 x.set_shape([None] * ndims)
1167 x.set_shape(x.get_shape().merge_with(shape))
1173 x.set_shape(x.get_shape().merge_with(shape))
1185 x.set_shape(x.get_shape().merge_with(shape))
|
| D | categorical.py | 58 event.set_shape(tensor_shape.TensorShape(params.get_shape()[:-1]))
316 ret.set_shape(self.batch_shape)
|
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| D | embedding_ops.py | 181 final_result.set_shape(tensor_shape.unknown_shape(
480 embeds.set_shape(ids.get_shape().concatenate(
673 weights.set_shape(
745 weights.set_shape(
860 partitioned_weight[p].set_shape(
|
| /external/tensorflow/tensorflow/contrib/distributions/python/ops/bijectors/ |
| D | softmax_centered.py | 149 y.set_shape(shape)
201 x.set_shape(shape)
|
| /external/tensorflow/tensorflow/contrib/solvers/python/ops/ |
| D | lanczos.py | 125 v.set_shape(v_shape)
161 r.set_shape(r_shape)
|
| /external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/state_space_models/ |
| D | state_space_model.py | 340 gathered_indices.set_shape(indices.get_shape().concatenate(
419 log_prob.set_shape(current_times.get_shape())
469 predicted_obs_var.set_shape(
472 predicted_obs.set_shape(current_times.get_shape().concatenate(
474 predicted_obs_var.set_shape(current_times.get_shape().concatenate(
|
| /external/tensorflow/tensorflow/contrib/tpu/ |
| D | tpu_estimator.md | 84 image.set_shape([28 * 28])
94 # set_shape to give inputs statically known shapes.
95 images.set_shape([batch_size, 28 * 28])
96 labels.set_shape([batch_size])
|
| /external/tensorflow/tensorflow/python/layers/ |
| D | core.py | 160 outputs.set_shape(output_shape)
377 outputs.set_shape(self.compute_output_shape(inputs.get_shape()))
|
123456