Home
last modified time | relevance | path

Searched defs:tensor_names (Results 1 – 25 of 27) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dintroduce_floating_point_jitter_pass_test.cc52 std::vector<string> tensor_names; in TEST() local
93 std::vector<string> tensor_names; in TEST() local
130 std::vector<string> tensor_names; in TEST() local
162 std::vector<string> tensor_names; in TEST() local
H A Dintroduce_floating_point_jitter_pass.cc28 const Graph& g, absl::Span<const string> tensor_names) { in GetNodesToModify()
123 absl::Span<string const> tensor_names, in IntroduceFloatingPointJitter()
H A Dflags.h158 std::vector<string> tensor_names; member
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DNamedTensorUtils.cpp237 auto tensor_names = tensor.names(); in compute_squeeze_outnames() local
251 auto tensor_names = tensor.names(); in compute_squeeze_outnames() local
268 auto tensor_names = tensor.names(); in compute_diagonal_outnames() local
418 auto tensor_names = tensor.names(); in broadcast_to_outnames() local
434 const auto tensor_names = tensor.names(); in compute_cat_outnames() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dsave_restore_v2_ops.cc44 const Tensor& prefix, const Tensor& tensor_names, in ValidateInputs()
99 const Tensor& tensor_names = context->input(1); in Compute() local
194 const Tensor& tensor_names = context->input(1); in Compute() local
H A Drestore_v2_op_test.cc66 const std::vector<string> tensor_names = { in RunTest() local
H A Dsave_restore_tensor.cc365 const Tensor& tensor_names, in RestoreTensorsV2()
H A Drestore_op_test.cc66 const std::vector<string> tensor_names = { in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/c/experimental/ops/
H A Dio_ops.cc51 AbstractTensorHandle* const tensor_names, in RestoreV2()
76 AbstractTensorHandle* const tensor_names, in SaveV2()
/aosp_15_r20/external/pytorch/torch/
H A D_namedtensor_internals.py67 def replace_ellipsis_by_position(ellipsis_idx, names, tensor_names): argument
74 def resolve_ellipsis(names, tensor_names, fn_name): argument
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/
H A Dsave_restore_spmd_expander.cc271 mlir::Value tensor_names = new_save.getArgument(1); in ConditionalSave() local
324 mlir::Value tensor_names = new_tensor_names[0]; in ConditionalSave() local
619 mlir::Value tensor_names = device_restore_fn.getArgument(1); in ExpandRestoreV2OpHelper() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DNamedTensor.cpp83 DimnameList tensor_names, in aligned_size()
193 const auto tensor_names = tensor.names(); in align_to() local
283 auto tensor_names = tensor.names(); in align_to() local
/aosp_15_r20/external/tensorflow/tensorflow/cc/tools/
H A Dfreeze_saved_model.cc38 std::unordered_set<string>* tensor_names) { in GetTensorNamesFromTensorInfo()
133 std::vector<string> tensor_names; in GetVariableNameToTensorMap() local
/aosp_15_r20/external/federated-compute/fcp/tensorflow/
H A Dappend_slices.py28 def append_slices(filename, tensor_names, shapes_and_slices, data, name=None): argument
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dquantize_training.cc190 Tensor tensor_names; in ConnectVariablesToSaveOp() local
274 Node* tensor_names; in AddRestoreVariableSubgraphs() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
H A Dconcat_xy.cc31 std::vector<std::string> tensor_names(op_def.src_tensors.size()); in GetConcatKernelCode() local
H A Dconcat_z.cc40 std::vector<std::string> tensor_names(op_def.src_tensors.size()); in GetConcatKernelCode() local
/aosp_15_r20/external/federated-compute/fcp/aggregation/testing/
H A Dtesting.cc60 tf::Status CreateTfCheckpoint(tf::Input filename, tf::Input tensor_names, in CreateTfCheckpoint()
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/
H A Ddtensor_sharded_prefix_kernel.cc38 const Tensor* tensor_names; in Compute() local
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dio_ops.py39 def _save(filename, tensor_names, tensors, tensor_slices=None, name="save"): argument
/aosp_15_r20/external/tensorflow/tensorflow/lite/python/
H A Dutil.py130 def get_tensors_from_tensor_names(graph, tensor_names): argument
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/
H A Dflatbuffer_import.cc938 const std::vector<std::string>& tensor_names) { in GetTensorIndices()
966 auto tensor_names = llvm::map_range( in BuildTFEntryFunctionAttribute() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/mobile/compatibility/
H A Dbackport_manager.cpp92 std::vector<std::string> tensor_names; in write_archive_current() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/serialization/
H A Dexport_module.cpp509 std::vector<std::string> tensor_names; in writeArchive() local
/aosp_15_r20/external/tensorflow/tensorflow/core/util/tensor_bundle/
H A Dtensor_bundle_test.cc940 std::vector<string> tensor_names = {"tensor-1-0", "tensor-0-1", "tensor-1-2", in TEST() local

12