Searched defs:output_specs (Results 1 – 13 of 13) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/experimental/rpc/ |
| H A D | rpc_ops.py | 220 output_specs=None, argument
367 def _add_method(self, method_name, output_specs, input_specs, client_handle, argument
411 output_specs=None, argument
442 def __init__(self, status_or, deleter, output_specs=None): argument
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/mobile/nnc/ |
| H A D | context.h | 155 const std::vector<OutputSpec>& output_specs() const { in output_specs() function
159 void set_output_specs(const std::vector<OutputSpec>& output_specs) { in set_output_specs()
|
| H A D | context.cpp | 179 std::vector<c10::IValue> output_specs; in serialize() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/distribute/experimental/rpc/kernels/ |
| H A D | rpc_ops.cc | 164 StructuredValue output_specs; member
185 const StructuredValue& output_specs) { in Register()
325 const StructuredValue& output_specs) { in Register()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/saved_model/ |
| H A D | saved_model.h | 60 output_specs; member
|
| H A D | saved_model.cc | 114 output_specs; in MapFunctionSignaturesFromTFSavedModelMLIR() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/distribute/experimental/rpc/proto/ |
| H A D | tf_rpc_service.proto | 22 StructuredValue output_specs = 3; field
|
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
| H A D | scatter_spmd_expander.cc | 41 std::vector<ShardingSpec> output_specs(tensor_rank); in GetOutputLayout() local
|
| H A D | gather_spmd_expander.cc | 314 std::vector<ShardingSpec> output_specs(params_rank - index_dimensions + in GatherNdGetOutputLayoutFromInput() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/saved_model/ |
| H A D | saved_model.h | 77 std::vector<TensorSpec> output_specs; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/ |
| H A D | quantization_context.cc | 143 llvm::SmallVector<Attribute, 4> output_specs; in Finalize() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/c/experimental/saved_model/core/ |
| H A D | saved_model_utils.cc | 173 std::vector<const TensorSpecProto*> output_specs; in ValidateSavedFunctionCompatibleWithFunctionDef() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/ |
| H A D | pmap_lib.cc | 540 const auto& output_specs = cache_entry.out_result_specs; in Call() local
|