Searched defs:set_shape (Results 1 – 12 of 12) sorted by relevance
| /external/ComputeLibrary/arm_compute/graph/ |
| D | TensorDescriptor.h | 64 TensorDescriptor &set_shape(TensorShape &tensor_shape) in set_shape() function
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_tensor_supported_values_test.py | 57 def set_shape(self, shape): member in WrappedTensor
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | state_ops.py | 37 def variable_op(shape, dtype, name="Variable", set_shape=True, container="", argument
|
| D | variables.py | 477 def set_shape(self, shape): member in Variable
1957 def set_shape(self, shape): member in RefVariable
|
| D | resource_variable_ops.py | 552 def set_shape(self, shape): member in BaseResourceVariable
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | keras_tensor.py | 268 def set_shape(self, shape): member in KerasTensor
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor.h | 716 void set_shape(const TensorShape& shape) { in set_shape() function
|
| D | tensor_test.cc | 36 static void set_shape(Tensor* t, const TensorShape& s) { t->set_shape(s); } in set_shape() function in tensorflow::TensorTestHelper
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | pjrt_c_api_client.cc | 639 void PjRtCApiBuffer::set_shape() { in set_shape() function in xla::PjRtCApiBuffer
|
| /external/tensorflow/tensorflow/python/keras/mixed_precision/ |
| D | autocast_variable.py | 182 def set_shape(self, shape): member in AutoCastVariable
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | ops.py | 690 def set_shape(self, shape): member in Tensor
1297 def set_shape(self, shape): member in _EagerTensorBase
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | backend.py | 4549 def set_shape(output_): function
|