Home
last modified time | relevance | path

Searched defs:component_shape (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/data/
H A Ddataset_utils.cc624 TensorShape component_shape(t.shape()); in ReadBatch() local
719 TensorShape component_shape((*batch)[i].shape()); in ProcessBatch() local
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/
H A Ddtensor_device.cc752 std::vector<int64_t>* component_shape, TF_Status* status) { in VerifyPackTensorShapeAndDtype()
838 std::vector<int64_t> component_shape; in Pack() local
879 std::vector<int64_t> component_shape; in Pack() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/
H A Dpadded_batch_dataset_op.cc343 TensorShape component_shape({}); in CopyBatch() local
/aosp_15_r20/external/tensorflow/tensorflow/c/eager/parallel_device/
H A Dparallel_device_lib.cc566 PartialTensorShape component_shape; in Shape() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dbarrier_ops.cc163 TensorShape component_shape(ready_tuples[0][i].shape()); in TryInsertMany() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/
H A Drebatch_dataset_op.cc509 TensorShape component_shape({batch_size}); in GetNextInternal() local
H A Dmap_and_batch_dataset_op.cc519 TensorShape component_shape({dataset()->batch_size_}); in EnsureOutputAllocated() local
/aosp_15_r20/cts/apps/CameraITS/utils/
H A Dopencv_processing_utils.py430 def component_shape(contour): function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Delemental_ir_emitter.cc2567 Shape component_shape = in MakeElementGenerator() local