Searched refs:set_shape (Results 1 – 25 of 138) sorted by relevance
123456
| /external/tensorflow/tensorflow/python/ops/ |
| D | string_ops.py | 229 indices.set_shape([None, 2])
230 values.set_shape([None])
231 shape.set_shape([2])
284 indices.set_shape([None, 2])
285 values.set_shape([None])
286 shape.set_shape([2])
|
| D | state_ops.py | 41 def variable_op(shape, dtype, name="Variable", set_shape=True, container="", argument
44 if not set_shape:
50 if set_shape:
51 ret.set_shape(shape)
|
| D | sparse_grad.py | 93 a_val_grad.set_shape(op.inputs[1].get_shape())
94 b_val_grad.set_shape(op.inputs[4].get_shape())
143 val_grad.set_shape(op.inputs[1].get_shape())
|
| D | rnn.py | 63 x_t.set_shape(
287 output.set_shape(flat_output.get_shape())
290 substate.set_shape(flat_substate.get_shape())
322 input_.set_shape(input_shape)
332 r.set_shape(input_shape)
845 input_.set_shape(shape[1:])
904 output.set_shape(shape)
1383 output.set_shape(tensor_shape.TensorShape(shape))
|
| D | data_flow_ops.py | 455 output.set_shape(shape)
497 output.set_shape(
539 output.set_shape(tensor_shape.TensorShape([None]).concatenate(shape))
1172 op.outputs[0].set_shape(tensor_shape.TensorShape([batch_dim])) # indices
1173 op.outputs[1].set_shape(tensor_shape.TensorShape([batch_dim])) # keys
1175 output.set_shape(
1406 out.set_shape(self._shape)
1794 output.set_shape(self._shapes[i])
|
| D | image_ops_impl.py | 325 result.set_shape([None, None, None])
327 result.set_shape(image_shape)
741 result.set_shape([None, None, image.get_shape()[2]])
773 result.set_shape([shape[0], None, None, shape[3]])
980 image.set_shape([
985 image.set_shape([
1061 image.set_shape([None] * 4)
1096 padded.set_shape(padded_shape)
1147 image.set_shape([None] * 4)
1179 cropped.set_shape(cropped_shape)
[all …]
|
| D | embedding_ops.py | 246 ret.set_shape(ids.get_shape().concatenate(element_shape_s))
537 weights.set_shape(
936 final_result.set_shape(
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | variable_ops_test.py | 77 p = state_ops.variable_op([1, 2], dtypes.float32, set_shape=False)
99 var = state_ops.variable_op(value.shape, dtypes.float32, set_shape=False)
107 var = state_ops.variable_op(value.shape, dtypes.float32, set_shape=False)
139 var = state_ops.variable_op([1, 2], dtypes.float32, set_shape=False)
148 var = state_ops.variable_op([1, 2], dtypes.float32, set_shape=False)
165 var = state_ops.variable_op([1, 2], dtypes.float32, set_shape=False)
181 var = state_ops.variable_op([1, 2], dtypes.float32, set_shape=False)
|
| D | tensordot_op_test.py | 123 a.set_shape([None, 2])
124 b.set_shape([2, 3])
133 a.set_shape([2, 2])
134 b.set_shape([2, None])
|
| D | neon_depthwise_conv_op_test.py | 121 t1.set_shape(tensor_in_sizes)
213 t1.set_shape(tensor_in_sizes)
|
| /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/python/ops/linalg/ |
| D | linear_operator_kronecker.py | 393 output.set_shape(broadcast_batch_shape.concatenate(
510 output.set_shape(broadcast_batch_shape.concatenate(
529 diag_part.set_shape(
552 product.set_shape(self.shape)
571 product.set_shape(self.shape[:-1])
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | dma_helper.h | 32 t->set_shape(s); in UnsafeSetShape()
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | SetSize.pbtxt | 12 name: "set_shape"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | SetSize.pbtxt | 12 name: "set_shape"
|
| /external/tensorflow/tensorflow/python/keras/layers/preprocessing/ |
| D | image_preprocessing.py | 281 output.set_shape(output_shape)
419 output.set_shape(inputs.shape)
566 output.set_shape(inputs.shape)
843 output.set_shape(inputs.shape)
988 output.set_shape(inputs.shape)
1105 output.set_shape(inputs.shape)
1197 output.set_shape(output_shape)
1294 output.set_shape(output_shape)
|
| /external/tensorflow/tensorflow/python/keras/layers/ops/ |
| D | core.py | 61 outputs.set_shape(output_shape)
|
| /external/tensorflow/tensorflow/python/ops/distributions/ |
| D | categorical.py | 57 event.set_shape(tensor_shape.TensorShape(params.shape[:-1]))
324 ret.set_shape(self.batch_shape)
|
| D | distribution.py | 1285 x.set_shape([None] * ndims)
1290 x.set_shape(x.get_shape().merge_with(shape))
1296 x.set_shape(x.get_shape().merge_with(shape))
1308 x.set_shape(x.get_shape().merge_with(shape))
|
| D | transformed_distribution.py | 447 log_prob.set_shape(
473 prob.set_shape(
572 entropy.set_shape(self.batch_shape)
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.-tensor.pbtxt | 64 name: "set_shape"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.-tensor.pbtxt | 64 name: "set_shape"
|
| /external/tensorflow/third_party/py/numpy/tf_numpy_api/ |
| D | tensorflow.experimental.numpy.ndarray.pbtxt | 64 name: "set_shape"
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | segment_id_ops.py | 135 splits.set_shape(tensor_shape.TensorShape([const_num_segments + 1]))
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | sparse_tensor.py | 378 indices.set_shape([None, rank])
395 dense_shape.set_shape([rank])
|
123456