Searched defs:output_specs (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/experimental/rpc/ |
| 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
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/saved_model/ |
| D | saved_model.h | 60 output_specs; member
|
| D | saved_model.cc | 114 output_specs; in MapFunctionSignaturesFromTFSavedModelMLIR() local
|
| /external/tensorflow/tensorflow/distribute/experimental/rpc/proto/ |
| D | tf_rpc_service.proto | 22 StructuredValue output_specs = 3; field
|
| /external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
| D | scatter_spmd_expander.cc | 41 std::vector<ShardingSpec> output_specs(tensor_rank); in GetOutputLayout() local
|
| D | gather_spmd_expander.cc | 314 std::vector<ShardingSpec> output_specs(params_rank - index_dimensions + in GatherNdGetOutputLayoutFromInput() local
|
| /external/tensorflow/tensorflow/core/tfrt/saved_model/ |
| D | saved_model.h | 77 std::vector<TensorSpec> output_specs; member
|
| /external/tensorflow/tensorflow/distribute/experimental/rpc/kernels/ |
| D | rpc_ops.cc | 164 StructuredValue output_specs; member
185 const StructuredValue& output_specs) { in Register()
325 const StructuredValue& output_specs) { in Register()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/quantization/ |
| D | quantization_context.cc | 143 llvm::SmallVector<Attribute, 4> output_specs; in Finalize() local
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/core/ |
| D | saved_model_utils.cc | 173 std::vector<const TensorSpecProto*> output_specs; in ValidateSavedFunctionCompatibleWithFunctionDef() local
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | pmap_lib.cc | 540 const auto& output_specs = cache_entry.out_result_specs; in Call() local
|