Home
last modified time | relevance | path

Searched defs:set_shape (Results 1 – 17 of 17) sorted by relevance

/aosp_15_r20/external/ComputeLibrary/arm_compute/graph/
H A DTensorDescriptor.h64 TensorDescriptor &set_shape(TensorShape &tensor_shape) in set_shape() function
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_cpp_lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/tensorflow/core/framework/
Dgraph_transfer_info.pb.h1937 inline void GraphTransferConstNodeInfo::set_shape(int index, ::int64_t value) { in set_shape() function
2233 inline void GraphTransferGraphInputNodeInfo::set_shape(int index, ::int64_t value) { in set_shape() function
2349 inline void GraphTransferGraphOutputNodeInfo::set_shape(int index, ::int64_t value) { in set_shape() function
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/ragged/
H A Dragged_tensor_supported_values_test.py57 def set_shape(self, shape): member in WrappedTensor
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dstate_ops.py37 def variable_op(shape, dtype, name="Variable", set_shape=True, container="", argument
H A Dvariables.py477 def set_shape(self, shape): member in Variable
1957 def set_shape(self, shape): member in RefVariable
H A Dresource_variable_ops.py552 def set_shape(self, shape): member in BaseResourceVariable
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/engine/
H A Dkeras_tensor.py268 def set_shape(self, shape): member in KerasTensor
/aosp_15_r20/prebuilts/clang/host/linux-x86/clang-r530567/python3/lib/python3.11/site-packages/pip/_vendor/rich/
Dsegment.py385 def set_shape( member in Segment
/aosp_15_r20/prebuilts/clang/host/linux-x86/clang-r530567b/python3/lib/python3.11/site-packages/pip/_vendor/rich/
Dsegment.py385 def set_shape( member in Segment
/aosp_15_r20/prebuilts/clang/host/linux-x86/clang-r522817/python3/lib/python3.11/site-packages/pip/_vendor/rich/
Dsegment.py385 def set_shape( member in Segment
/aosp_15_r20/prebuilts/clang/host/linux-x86/clang-r536225/python3/lib/python3.11/site-packages/pip/_vendor/rich/
Dsegment.py385 def set_shape( member in Segment
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dtensor.h716 void set_shape(const TensorShape& shape) { in set_shape() function
H A Dtensor_test.cc36 static void set_shape(Tensor* t, const TensorShape& s) { t->set_shape(s); } in set_shape() function in tensorflow::TensorTestHelper
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dpjrt_c_api_client.cc639 void PjRtCApiBuffer::set_shape() { in set_shape() function in xla::PjRtCApiBuffer
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/mixed_precision/
H A Dautocast_variable.py182 def set_shape(self, shape): member in AutoCastVariable
/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Dops.py690 def set_shape(self, shape): member in Tensor
1297 def set_shape(self, shape): member in _EagerTensorBase
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/
H A Dbackend.py4549 def set_shape(output_): function