Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/ops/
H A Dcompression_ops.py35 def uncompress(element, output_spec): argument
/aosp_15_r20/external/pytorch/torch/_library/
H A Dautograd.py118 output_spec: Optional[spec_t] = None variable in supports_tensorlist.Metadata
/aosp_15_r20/external/pytorch/test/mobile/nnc/
H A Dtest_context.cpp46 OutputSpec output_spec; in create_test_output_spec() local
/aosp_15_r20/external/googleapis/google/cloud/aiplatform/v1beta1/
H A Dmodel_monitoring_spec.proto44 ModelMonitoringOutputSpec output_spec = 3; field
/aosp_15_r20/external/pytorch/torch/distributed/tensor/
H A D_op_schema.py92 def output_spec(self) -> DTensorSpec: member in PlacementStrategy
/aosp_15_r20/external/pytorch/torch/csrc/jit/mobile/nnc/
H A Dcontext.cpp181 for (const auto& output_spec : output_specs_) { in serialize() local
/aosp_15_r20/external/tensorflow/tensorflow/c/
H A Dops.cc50 const char* output_spec) { in TF_OpDefinitionBuilderAddOutput()
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/kernel/
H A Dtfrt_op_kernel.cc220 TFRTOpMetaBuilder& TFRTOpMetaBuilder::Output(StringPiece output_spec) { in Output()
/aosp_15_r20/external/linux-kselftest/tools/testing/selftests/bpf/
H A Dveristat.c78 struct stat_specs output_spec; member