- Sort Score
- Result 10 results
- Languages All
Results 1 - 7 of 7 for dyn_cast (0.05 sec)
-
tensorflow/c/eager/c_api_unified_experimental.cc
// // ============================================================================= using tensorflow::AbstractFunction; using tensorflow::AbstractTensorHandle; using tensorflow::DataType; using tensorflow::dyn_cast; using tensorflow::OutputList; using tensorflow::Status; using tensorflow::unwrap; using tensorflow::wrap; using tensorflow::tracing::CreateTracingExecutionContext;
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 05:11:17 UTC 2024 - 9K bytes - Viewed (0) -
tensorflow/c/eager/c_api_unified_experimental_graph.cc
return errors::Internal("Expected 1 output but found ", num_outputs); } auto* t = dyn_cast<GraphTensor>(outputs[0]); if (!t) { return tensorflow::errors::InvalidArgument( "Unable to cast input to GraphTensor"); } inputs_.push_back(t->output_); *output = tensorflow::down_cast<TracingTensorHandle*>(outputs[0]); return absl::OkStatus(); }
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 05:11:17 UTC 2024 - 15.7K 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);
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 05:11:17 UTC 2024 - 7K bytes - Viewed (0) -
tensorflow/c/eager/gradients.cc
AbstractOperationPtr op(ctx->CreateOperation()); TF_RETURN_IF_ERROR(op->Reset("ZerosLike", /*raw_device_name=*/nullptr)); if (isa<tracing::TracingOperation>(op.get())) { TF_RETURN_IF_ERROR(dyn_cast<tracing::TracingOperation>(op.get())->SetOpName( absl::StrCat("ZerosLike", ToId(t)).c_str())); } TF_RETURN_IF_ERROR(op->AddInput(t)); int num_outputs = 1;
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 05:11:17 UTC 2024 - 19.7K bytes - Viewed (0) -
tensorflow/c/eager/c_api_experimental.cc
// One of the inputs we're trying to pack is on a custom device. We'll let // the first custom device we see handle all of the packing. auto* custom_device_handle = tensorflow::down_cast<tensorflow::CustomDeviceTensorHandle*>( unwrapped_handle); tensorflow::ImmediateExecutionTensorHandle* result; status->status = custom_device_handle->device()->Pack(
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 05:11:17 UTC 2024 - 35.9K bytes - Viewed (0) -
tensorflow/c/eager/c_api.cc
tensorflow::unwrap(h); // TODO(b/175427838): It would be nice to be able to use tensorflow::isa here. if (tensorflow::CustomDeviceTensorHandle::classof(unwrapped_handle)) { return tensorflow::down_cast<tensorflow::CustomDeviceTensorHandle*>( unwrapped_handle) ->DevicePointer(); } // TODO(b/175427838): It would be nice to be able to use tensorflow::isa here.
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 05:11:17 UTC 2024 - 43.9K bytes - Viewed (0) -
tensorflow/c/c_api.cc
tensorflow::TensorProto from_tensor_proto; status->status = BufferToMessage(from, &from_tensor_proto); if (!status->status.ok()) { return; } status->status = tensorflow::down_cast<tensorflow::TensorInterface*>(to->tensor) ->FromProto(from_tensor_proto); } // --------------------------------------------------------------------------
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 16:27:48 UTC 2024 - 102.3K bytes - Viewed (0)