| /external/tensorflow/tensorflow/python/ops/ |
| D | string_ops.py | 227 indices.set_shape([None, 2])
228 values.set_shape([None])
229 shape.set_shape([2])
282 indices.set_shape([None, 2])
283 values.set_shape([None])
284 shape.set_shape([2])
|
| D | state_ops.py | 37 def variable_op(shape, dtype, name="Variable", set_shape=True, container="", argument
40 if not set_shape:
46 if set_shape:
47 ret.set_shape(shape)
|
| D | rnn.py | 59 x_t.set_shape(
286 output.set_shape(flat_output.get_shape())
289 substate.set_shape(flat_substate.get_shape())
321 input_.set_shape(input_shape)
331 r.set_shape(input_shape)
909 input_.set_shape(shape[1:])
968 output.set_shape(shape)
1449 output.set_shape(tensor_shape.TensorShape(shape))
|
| D | sparse_grad.py | 86 a_val_grad.set_shape(op.inputs[1].get_shape())
87 b_val_grad.set_shape(op.inputs[4].get_shape())
136 val_grad.set_shape(op.inputs[1].get_shape())
|
| D | data_flow_ops.py | 457 output.set_shape(shape)
499 output.set_shape(
541 output.set_shape(tensor_shape.TensorShape([None]).concatenate(shape))
1173 op.outputs[0].set_shape(tensor_shape.TensorShape([batch_dim])) # indices
1174 op.outputs[1].set_shape(tensor_shape.TensorShape([batch_dim])) # keys
1176 output.set_shape(
1407 out.set_shape(self._shape)
1795 output.set_shape(self._shapes[i])
|
| D | image_ops_impl.py | 322 result.set_shape([None, None, None])
324 result.set_shape(image_shape)
739 result.set_shape([None, None, image.get_shape()[2]])
771 result.set_shape([shape[0], None, None, shape[3]])
980 image.set_shape([
985 image.set_shape([
1108 image.set_shape([None] * 4)
1144 padded.set_shape(padded_shape)
1212 image.set_shape([None] * 4)
1245 cropped.set_shape(cropped_shape)
[all …]
|
| /external/tensorflow/tensorflow/python/kernel_tests/variables/ |
| D | variable_ops_test.py | 73 p = state_ops.variable_op([1, 2], dtypes.float32, set_shape=False)
96 var = state_ops.variable_op(value.shape, dtypes.float32, set_shape=False)
104 var = state_ops.variable_op(value.shape, dtypes.float32, set_shape=False)
136 var = state_ops.variable_op([1, 2], dtypes.float32, set_shape=False)
145 var = state_ops.variable_op([1, 2], dtypes.float32, set_shape=False)
162 var = state_ops.variable_op([1, 2], dtypes.float32, set_shape=False)
178 var = state_ops.variable_op([1, 2], dtypes.float32, set_shape=False)
|
| /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/security/advisory/ |
| D | tfsa-2022-127.md | 7 When `SetSize` receives an input `set_shape` that is not a 1D tensor, it gives a `CHECK` fails that…
15 tf.raw_ops.SetSize(set_indices=arg_0, set_values=arg_1, set_shape=arg_2,
|
| /external/tensorflow/tensorflow/python/kernel_tests/math_ops/ |
| D | tensordot_op_test.py | 119 a.set_shape([None, 2])
120 b.set_shape([2, 3])
129 a.set_shape([2, 2])
130 b.set_shape([2, None])
|
| /external/tensorflow/tensorflow/python/ops/linalg/ |
| D | linear_operator_kronecker.py | 363 output.set_shape(broadcast_batch_shape.concatenate(
435 diag_part.set_shape(
458 product.set_shape(self.shape)
477 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/ComputeLibrary/src/graph/nodes/ |
| D | ReductionLayerNode.cpp | 79 output_info.set_shape(output_shape); in configure_output()
|
| D | ArgMinMaxLayerNode.cpp | 89 output_info.set_shape(output_shape); in configure_output()
|
| /external/ComputeLibrary/arm_compute/graph/ |
| D | TensorDescriptor.h | 64 TensorDescriptor &set_shape(TensorShape &tensor_shape) in set_shape() function
|
| /external/tensorflow/tensorflow/python/ops/distributions/ |
| D | categorical.py | 53 event.set_shape(tensor_shape.TensorShape(params.shape[:-1]))
322 ret.set_shape(self.batch_shape)
|
| D | distribution.py | 1278 x.set_shape([None] * ndims)
1283 x.set_shape(x.get_shape().merge_with(shape))
1289 x.set_shape(x.get_shape().merge_with(shape))
1301 x.set_shape(x.get_shape().merge_with(shape))
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | function_def_to_graph.py | 123 op.outputs[output_index].set_shape(shape)
313 tensor.set_shape([])
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_tensor.py | 1542 self._row_partition._row_splits.set_shape(shape[0] + 1)
1567 self.flat_values.set_shape(flat_shape)
1740 has_default.set_shape(tensor_shape.TensorShape([None, None]))
1741 has_default.set_shape(tensor.shape[:2])
1858 tensor.set_shape(output_shape)
2524 result.flat_values.set_shape(self.flat_values_spec.shape)
2526 result.set_shape(self._shape)
2598 result.flat_values.set_shape(self.flat_values_spec.shape)
2600 result.set_shape(self._shape)
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.-tensor.pbtxt | 64 name: "set_shape"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.-tensor.pbtxt | 64 name: "set_shape"
|
| D | tensorflow.__internal__.tracking.-checkpoint-initial-value.pbtxt | 69 name: "set_shape"
|
| /external/tensorflow/third_party/py/numpy/tf_numpy_api/ |
| D | tensorflow.experimental.numpy.ndarray.pbtxt | 64 name: "set_shape"
|