- Sort Score
- Result 10 results
- Languages All
Results 1 - 8 of 8 for Kast (0.17 sec)
-
tensorflow/c/eager/c_api_unified_experimental.cc
Status status = tensorflow::PartialTensorShape::MakePartialShape( reinterpret_cast<int64_t*>(shape.dim_sizes), shape.num_dims, &partial_shape); if (!status.ok()) { tsl::Set_TF_Status_from_Status(s, status); return nullptr; } } tsl::Set_TF_Status_from_Status( s, tracing_ctx->AddParameter(static_cast<DataType>(dtype), partial_shape, &t));
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 9K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/nn_grad.cc
if (!isa<ImmediateExecutionContext>(ctx)) { // TODO(b/168850692): Fix this. return errors::Unimplemented( "BroadcastMul is not supported in tracing mode yet."); } auto imm_ctx = dyn_cast<ImmediateExecutionContext>(ctx); AbstractTensorPtr minus_1(imm_ctx->CreateInt32Scalar(-1)); ImmediateTensorHandlePtr dim(imm_ctx->CreateLocalHandle(minus_1.get())); AbstractTensorHandle* expand_dims_outputs;
C++ - Registered: Tue Mar 26 12:39:09 GMT 2024 - Last Modified: Wed Feb 28 13:53:47 GMT 2024 - 5.7K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/custom_gradient_test.cc
ASSERT_EQ(errors::OK, s.code()) << s.message(); TF_Tensor* result_tensor; s = GetValue(outputs[0], &result_tensor); ASSERT_EQ(errors::OK, s.code()) << s.message(); auto result_value = static_cast<float*>(TF_TensorData(result_tensor)); EXPECT_EQ(*result_value, 1.0); outputs[0]->Unref(); TF_DeleteTensor(result_tensor); result_tensor = nullptr; } INSTANTIATE_TEST_SUITE_P(
C++ - Registered: Tue Mar 26 12:39:09 GMT 2024 - Last Modified: Wed Feb 28 13:53:47 GMT 2024 - 4.8K bytes - Viewed (0) -
tensorflow/c/eager/immediate_execution_tensor_handle.cc
std::string& summary) const { Status status; AbstractTensorPtr resolved( // TODO(allenl): Resolve should be const, and the caches that get updated // marked mutable. const_cast<ImmediateExecutionTensorHandle*>(this)->Resolve(&status)); if (!status.ok()) { return status; } summary = resolved->SummarizeValue(); return absl::OkStatus(); }
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 2.1K bytes - Viewed (0) -
tensorflow/c/eager/gradients_test.cc
/*raw_device_name=*/nullptr, &forward_op); ASSERT_EQ(errors::OK, s.code()) << s.message(); if (isa<TracingOperation>(check_numerics_op.get())) { s = dyn_cast<TracingOperation>(check_numerics_op.get()) ->SetOpName("check_numerics"); ASSERT_EQ(errors::OK, s.code()) << s.message(); } s = AddInput(check_numerics_op.get(), t.get(), &forward_op);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 7K bytes - Viewed (0) -
tensorflow/c/eager/tracing_utils.cc
namespace tracing { Status MaybeSetOpName(AbstractOperation* op, const char* op_name) { if (isa<TracingOperation>(op)) { TF_RETURN_IF_ERROR(dyn_cast<TracingOperation>(op)->SetOpName(op_name)); } if (isa<gradients::TapeOperation>(op)) { TF_RETURN_IF_ERROR(MaybeSetOpName( dyn_cast<gradients::TapeOperation>(op)->GetBackingOperation(), op_name)); } return absl::OkStatus(); } } // namespace tracing
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Feb 27 13:57:45 GMT 2024 - 1.4K bytes - Viewed (0) -
tensorflow/c/eager/unified_api_test.cc
} AbstractContextPtr ctx(BuildFunction("test_fn")); AbstractTensorHandlePtr x; { tracing::TracingTensorHandle* x_raw = nullptr; PartialTensorShape shape; Status s = dyn_cast<tracing::TracingContext>(ctx.get())->AddParameter( DT_FLOAT, shape, &x_raw); ASSERT_EQ(errors::OK, s.code()) << s.message(); x.reset(x_raw); } PartialTensorShape shape;
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Feb 27 13:57:45 GMT 2024 - 6.7K bytes - Viewed (0) -
tensorflow/c/eager/unified_api_testutil.cc
tracing::TracingTensorHandle* handle = nullptr; for (auto input : inputs) { PartialTensorShape shape; TF_RETURN_IF_ERROR(input->Shape(&shape)); TF_RETURN_IF_ERROR(dyn_cast<tracing::TracingContext>(ctx)->AddParameter( input->DataType(), shape, &handle)); params->emplace_back(handle); } return absl::OkStatus(); } // Runs `model` maybe wrapped in a function.
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Feb 27 13:57:45 GMT 2024 - 5.7K bytes - Viewed (0)