- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 14 for tensor_names (0.2 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_save_op.cc
} // Creates a 1D string array constant for "tensor_names" input of `RestoreV2` // op. The `ConstOp` will be created at `builder`'s current insertion point. TF::ConstOp CreateTensorNamesConst(const ArrayRef<std::string> tensor_names, OpBuilder& builder) { const auto loc = NameLoc::get(builder.getStringAttr("tensor_names")); return Create1DStringConst(tensor_names, loc, builder); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 22 05:52:39 UTC 2024 - 9.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_import.cc
const std::vector<std::string>& tensor_names) { absl::flat_hash_map<std::string, int> name_to_index; for (const auto& index_and_tensor : llvm::enumerate(subgraph.tensors)) { name_to_index[index_and_tensor.value()->name] = index_and_tensor.index(); } std::vector<int> indices; indices.reserve(tensor_names.size()); for (const auto& name : tensor_names) { auto found = name_to_index.find(name);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 18:21:50 UTC 2024 - 66.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/host_runtime/tfrt_ops.td
let arguments = (ins Arg<TF_StrTensor, [{Must have a single element. The prefix of a V2 checkpoint.}]>:$prefix, Arg<TF_StrTensor, [{shape {N}. The names of the tensors to be restored. This is tensor name in the model.}]>:$tensor_names, Arg<TF_StrTensor, [{shape {N}. The slice specs of the tensors to be restored. Empty strings indicate that they are non-partitioned tensors.}]>:$shape_and_slices, // VarHandles for restored tensors.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 31 20:44:15 UTC 2024 - 6.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/ir/mlrt/tf_mlrt_ops.td
because the tensors are loaded asynchronously. }]; let arguments = (ins TFTensorType:$prefix, TFTensorType:$tensor_names, TFTensorType:$shape_and_slices, Variadic<TFTensorType>:$var_handles, TypeArrayAttr: $restored_dtypes, DenseBoolArrayAttr: $truncate_in_cast ); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 31 20:44:15 UTC 2024 - 13.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 23:24:08 UTC 2024 - 793K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/cc/convert_asset_args.cc
} } AssetFileDef CreateAssetFileDef(const StringRef filename, const StringRef tensor_name) { AssetFileDef asset_file_def{}; asset_file_def.set_filename(MaybeStripAssetDirectoryPrefix(filename).str()); tensorflow::TensorInfo tensor_info{}; tensor_info.set_name(tensor_name.str()); *asset_file_def.mutable_tensor_info() = tensor_info; return asset_file_def; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.4K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_host_recv_device_context.cc
#include "tensorflow/compiler/tf2xla/shape_util.h" #include "tensorflow/compiler/tf2xla/type_util.h" namespace tensorflow { void XlaHostRecvDeviceContext::CopyDeviceTensorToCPU( const Tensor* device_tensor, StringPiece tensor_name, Device* device, Tensor* cpu_tensor, StatusCallback done) { DataType dtype = EncodePrimitiveTypeAsDataType(shape_.element_type()).value(); TensorShape tensor_shape;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 22:46:36 UTC 2024 - 1.9K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_host_send_device_context.h
bool sync_dst_compute) const override; void CopyDeviceTensorToCPU(const Tensor* device_tensor, StringPiece tensor_name, Device* device, Tensor* cpu_tensor, StatusCallback done) override { done(errors::Internal("host->device copy not implemented.")); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 22:46:36 UTC 2024 - 3.7K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_host_recv_device_context.h
} // Copies `device_memory_base_` with `shape_` into `cpu_tensor`. // `device_tensor` is unused. void CopyDeviceTensorToCPU(const Tensor* device_tensor, StringPiece tensor_name, Device* device, Tensor* cpu_tensor, StatusCallback done) override; void CopyTensorInSameDevice(const Tensor* input_tensor, Device* device, Tensor* output_tensor,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 22:46:36 UTC 2024 - 3.9K bytes - Viewed (0) -
tensorflow/compiler/jit/pjrt_device_context.cc
} else { return first_try_buffer.status(); } } } // namespace void PjRtDeviceContext::CopyDeviceTensorToCPU(const Tensor* device_tensor, absl::string_view tensor_name, Device* device, Tensor* cpu_tensor, StatusCallback done) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 13 08:49:31 UTC 2024 - 11.6K bytes - Viewed (0)