- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 136 for tensor_b (0.29 sec)
-
tensorflow/c/eager/tape.h
Gradient* grad_tensor = grad[grad_index]; if (grad_tensor != nullptr) { int64_t tensor_id = vspace_.TensorId(grad_tensor); targets.push_back(tensor_id); if (sources_set.find(tensor_id) != sources_set.end()) { sources_that_are_targets.emplace( tensor_id, vspace_.TapeTensorFromGradient(grad_tensor)); } Gradient* in_grad = in_grads[grad_index];
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 02 12:40:29 UTC 2024 - 47.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/remat/rematerializer.cc
for (const int itensor : operations_[ioperation].tensors) { if (const auto& tensor = tensors_[itensor]; tensor.first_use() >= begin && tensor.first_use() < end && tensor.last_use() > peak_loc) { for (const int ioperation : tensor.operations) { // We return the earliest dependence on any output tensor. if (ioperation > peak_loc && ioperation < best) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 14 20:57:44 UTC 2023 - 13.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_op_base.td
// Tensor types def TF_BoolTensor : TensorOf<[TF_Bool]>; def TF_IntTensor : TensorOf<[TF_Int]>; def TF_Int4Tensor : TensorOf<[TF_Int4]>; def TF_Int8Tensor : TensorOf<[TF_Int8]>; def TF_Int16Tensor : TensorOf<[TF_Int16]>; def TF_Int32Tensor : TensorOf<[TF_Int32]>; def TF_Int64Tensor : TensorOf<[TF_Int64]>; def TF_I32OrI64Tensor : TensorOf<[TF_I32OrI64]>;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 30.5K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_lib.h
std::vector<TFE_TensorHandle*> result; result.reserve(tensors_.size()); for (const TensorHandlePtr& tensor : tensors_) { result.emplace_back(tensor.get()); } return result; } private: ParallelTensor(const ParallelDevice& device, std::vector<TensorHandlePtr> tensors, absl::Span<const int64_t> shape, const TF_DataType dtype) : device_(device),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 25 15:21:13 UTC 2023 - 12.9K bytes - Viewed (0) -
tensorflow/c/c_api_experimental.h
TF_Status* status); // On success, enqueues `tensor` into a TF-managed FifoQueue given by // `tensor_id`, associated with `session`. There must be a graph node named // "fifo_queue_enqueue_<tensor_id>", to be executed by this API call. It reads // from a placeholder node "arg_tensor_enqueue_<tensor_id>". //
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 27 21:07:00 UTC 2023 - 15.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/test_schema.fbs
// model. table SubGraph { // A list of all tensors used in this subgraph. tensors:[Tensor]; // Indices of the tensors that are inputs into this subgraph. Note this is // the list of non-static tensors that feed into the subgraph for inference. inputs:[int]; // Indices of the tensors that are outputs out of this subgraph. Note this is // the list of output tensors that are considered the product of the
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 19 19:46:06 UTC 2021 - 26.1K bytes - Viewed (0) -
tensorflow/c/tf_tensor.cc
auto* ret = emptyTensor->tensor; delete emptyTensor; return ret; } Tensor tensor; if (!tensor.CopyFrom(src, src.shape())) { return nullptr; } return new tensorflow::TensorInterface(std::move(tensor)); } // Non-static for testing. TF_Tensor* TF_TensorFromTensor(const tensorflow::Tensor& src, Status* status) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Apr 14 21:57:32 UTC 2024 - 11.5K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_lib.cc
for (int component_index = 0; component_index < tensors_.size(); ++component_index) { // TODO(allenl): Add a C API for summarizing tensors. Currently custom // devices limiting themselves to a C API (for ABI compatibility) would need // to implement summarization for component tensors themselves. ImmediateExecutionTensorHandle* component = tensorflow::unwrap(tensors_[component_index].get());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Feb 09 07:47:20 UTC 2024 - 25.4K bytes - Viewed (0) -
tensorflow/cc/framework/ops.h
DataType data_type() const { return data_type_; } Status status() const { return status_; } const Tensor& tensor() const { return tensor_; } private: Status status_; Output output_ = Output(Operation(nullptr), 0); Tensor tensor_; const std::string node_name_ = ""; int32 index_ = 0; DataType data_type_ = DT_INVALID; };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 13 05:57:22 UTC 2024 - 10.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/lower_tf.td
(TF_SubOp $max_range, $min_range), (TF_ConstOp (GetF32Scalar<255>)))), $min_range), /*truncate=*/ConstBoolAttrFalse), [(TensorOf<[TF_Qint8, TF_Quint8]> $input), (TF_Float32Tensor $min_range), (TF_Float32Tensor $max_range), (TensorOf<[TF_Float32, TF_Bfloat16]> $result)]>; //===----------------------------------------------------------------------===// // Difference op patterns.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 04 13:30:42 UTC 2024 - 24.7K bytes - Viewed (0)