- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 36 for rack (0.15 sec)
-
tensorflow/c/c_api.cc
} tensorflow::shape_inference::ShapeHandle shape = ic->output(output.index); int rank = -1; if (ic->RankKnown(shape)) { rank = ic->Rank(shape); } if (num_dims != rank) { status->status = InvalidArgument("Expected rank is ", num_dims, " but actual rank is ", rank); return; } if (num_dims == 0) { // Output shape is a scalar.
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 102.3K bytes - Viewed (0) -
tensorflow/c/eager/c_api_unified_experimental.cc
return wrap(unwrap(o)->outputs[i]); } void TF_OutputListPushBack(TF_OutputList* o, TF_AbstractTensor* tensor, TF_Status* s) { unwrap(o)->outputs.push_back(unwrap(tensor)); } void TF_AbstractOpSetOpType(TF_AbstractOp* op, const char* const op_type, TF_Status* s) { tsl::Set_TF_Status_from_Status( s, unwrap(op)->Reset(op_type,
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 9K bytes - Viewed (0) -
tensorflow/c/eager/gradients.cc
forward_op_->inputs.push_back(input); return absl::OkStatus(); } Status AddInputList(AbstractOperation* op_, absl::Span<AbstractTensorHandle* const> inputs, ForwardOperation* forward_op_) { TF_RETURN_IF_ERROR(op_->AddInputList(inputs)); for (auto input : inputs) { forward_op_->inputs.push_back(input); } return absl::OkStatus(); }
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 19.3K bytes - Viewed (0) -
tensorflow/c/c_test_util.cc
inputs_.clear(); for (const auto& p : inputs) { inputs_.emplace_back(TF_Output{p.first, 0}); input_values_.emplace_back(p.second); } } void CSession::SetOutputs(std::initializer_list<TF_Operation*> outputs) { ResetOutputValues(); outputs_.clear(); for (TF_Operation* o : outputs) { outputs_.emplace_back(TF_Output{o, 0}); } output_values_.resize(outputs_.size()); }
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Fri Oct 15 03:16:52 GMT 2021 - 17.8K bytes - Viewed (2) -
tensorflow/c/experimental/gradients/array_grad_test.cc
}; TEST_P(CppGradients, TestIdentityNGrad) { // This test is interesting because the current implementation of GradientTape // would return [0, 1] whereas we use build_default_zeros_grads=false here // so we get back [nullptr, 1]. AbstractTensorHandlePtr x1; { AbstractTensorHandle* x1_raw = nullptr; status_ = TestScalarTensorHandle<float, TF_FLOAT>( immediate_execution_ctx_.get(), 1.0f, &x1_raw);
C++ - Registered: Tue Mar 26 12:39:09 GMT 2024 - Last Modified: Wed Feb 28 13:53:47 GMT 2024 - 5K bytes - Viewed (0) -
tensorflow/c/c_api_experimental.cc
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 29.4K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device.cc
return result; } components.emplace_back(TFE_TensorHandleCopySharingTensor( absl::get<TFE_TensorHandle*>(inputs[i]), status)); } std::vector<MaybeParallelTensorOwned> result_content; result_content.reserve(1); result_content.push_back(ParallelTensor::FromTensorHandles( parallel_device, std::move(components), status));
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Wed Mar 29 22:05:31 GMT 2023 - 18.3K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_lib.cc
components.reserve(underlying_devices_.size()); for (int j = 0; j < underlying_devices_.size(); ++j) { components.push_back(std::move(per_device_output_tensors[j][i])); } if (expected_output_shapes[i].IsFullyDefined()) { per_device_outputs.push_back(ParallelTensor::FromTensorHandles( *this, std::move(components), absl::Span<const int64_t>(expected_output_shapes[i].dim_sizes()),
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Fri Feb 09 07:47:20 GMT 2024 - 25.4K bytes - Viewed (1) -
tensorflow/c/eager/parallel_device/parallel_device_lib_test.cc
ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get()); std::vector<TensorHandlePtr> vector_handles; vector_handles.reserve(2); vector_handles.push_back(std::move(two_vector)); vector_handles.push_back(std::move(three_vector)); std::unique_ptr<ParallelTensor> unknown_length_vector = ParallelTensor::FromTensorHandles( parallel_device, std::move(vector_handles), status.get());
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Thu Jul 08 23:47:35 GMT 2021 - 15.3K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_testlib.cc
TF_NewTensor(TF_FLOAT, nullptr, 0, values, num_bytes, &FloatDeallocator, nullptr), TF_DeleteTensor); return TensorHandlePtr(TFE_NewTensorHandle(tensor.get(), status)); } // Creates a rank-one TFE_TensorHandle with value `v`. TensorHandlePtr VectorFloatTensorHandle(const std::vector<float>& v, TF_Status* status) { const int num_bytes = v.size() * sizeof(float);
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Tue Jun 15 15:44:44 GMT 2021 - 12.5K bytes - Viewed (0)