Home
last modified time | relevance | path

Searched defs:InputDescriptor (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/
H A Drebatch_dataset_op.cc246 struct InputDescriptor { struct in tensorflow::data::experimental::__anondafc533c0111::RebatchDatasetOp::Dataset::Iterator
247 InputDescriptor() {} in InputDescriptor() argument
248 InputDescriptor(Tensor&& whole_tensor, int64_t original_batch_dim, in InputDescriptor() function
254 Tensor whole_tensor;
255 int64_t original_batch_dim;
256 int64_t interval;
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dsubgraph_lowering.py79 class InputDescriptor: class
105 subgraph: ir.Subgraph, inputs: List[InputDescriptor]