- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 56 for init (0.19 sec)
-
tensorflow/c/eager/parallel_device/parallel_device_lib.cc
} TFE_OpAddAttrs(op_.get(), attributes); for (int input_index = 0; input_index < inputs.size(); ++input_index) { TFE_OpAddInput(op_.get(), inputs[input_index], status); if (TF_GetCode(status) != TF_OK) return; } std::vector<TFE_TensorHandle*> unwrapped_results(expected_max_outputs); int real_num_outputs = expected_max_outputs;
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Feb 09 07:47:20 GMT 2024 - 25.4K bytes - Viewed (1) -
tensorflow/c/eager/gradient_checker.cc
for (int32_t i = start; i < end; i += step) { (*data)[i] = i; } } // Fills out_dims with the dimensions of the given tensor. void GetDims(const TF_Tensor* t, int64_t* out_dims) { int num_dims = TF_NumDims(t); for (int i = 0; i < num_dims; i++) { out_dims[i] = TF_Dim(t, i); } } // Runs model as is if output is a scalar, // else sums the output tensor before returning.
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 7.3K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_testlib.cc
TFE_OpSetDevice(op.get(), device, status); if (TF_GetCode(status) != TF_OK) return; TFE_TensorHandle* result_handles[num_replicas]; int num_retvals = num_replicas; TFE_Execute(op.get(), result_handles, &num_retvals, status); if (TF_GetCode(status) != TF_OK) return; for (int i = 0; i < num_replicas; ++i) { (*components)[i].reset(result_handles[i]); } }
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Jun 15 15:44:44 GMT 2021 - 12.5K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device.cc
reinterpret_cast<NamedParallelDevice*>(device_info); std::vector<MaybeParallelTensorUnowned> typed_inputs; int num_inputs = TFE_OpGetFlatInputCount(original_op, status); if (TF_GetCode(status) != TF_OK) return; typed_inputs.reserve(num_inputs); for (int i = 0; i < num_inputs; ++i) { TFE_TensorHandle* input = TFE_OpGetFlatInput(original_op, i, status); if (TF_GetCode(status) != TF_OK) return;
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Mar 29 22:05:31 GMT 2023 - 18.3K bytes - Viewed (0) -
tensorflow/c/eager/custom_device_testutil.cc
~LoggedTensor() { TFE_DeleteTensorHandle(tensor); } }; int64_t LoggedTensorDim(void* data, int dim_index, TF_Status* status) { return TFE_TensorHandleDim(reinterpret_cast<LoggedTensor*>(data)->tensor, dim_index, status); } int LoggedTensorNumDims(void* data, TF_Status* status) { return TFE_TensorHandleNumDims(reinterpret_cast<LoggedTensor*>(data)->tensor,
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Mar 03 20:47:31 GMT 2021 - 8.3K bytes - Viewed (0) -
tensorflow/c/experimental/next_pluggable_device/c_api.cc
// by the underlying tensorflow::VariableInfo. TF_VariableInfo(int index, const std::string& name, tensorflow::Var* var) { var_info = tensorflow::VariableInfo{index, name, var}; } tensorflow::VariableInfo var_info{0, "", nullptr}; }; TF_VariableInfo* TF_CreateVariableInfoFromContext(TF_OpKernelContext* ctx, int index,
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Tue Jan 09 00:52:04 GMT 2024 - 13.9K bytes - Viewed (1) -
tensorflow/c/eager/parallel_device/parallel_device_test.cc
TFE_TensorHandle* result_handles; int num_retvals = 1; TFE_Execute(op.get(), &result_handles, &num_retvals, status.get()); ASSERT_TRUE(TF_GetCode(status.get()) == TF_INVALID_ARGUMENT) << TF_Message(status.get()); } } TensorHandlePtr CollectiveSum(TFE_Context* context, TFE_TensorHandle* input, int group_size, TF_Status* status) {
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Jul 08 23:47:35 GMT 2021 - 29.3K bytes - Viewed (1) -
tensorflow/c/eager/c_api_distributed_test.cc
// When remote is false, function device is placed on task0. Handle types are // REMOTE, REMOTE, LOCAL on task0. When remote is true, function device is // placed on task1, Handle types are LOCAL, REMOTE, LOCAL on task1. int num_replicas = 3; std::vector<TFE_TensorHandle*> handles = {h0, h1, h2}; TFE_TensorHandle* packed_handle = TFE_CreatePackedTensorHandle(ctx, handles.data(), &num_replicas, status);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 23.5K bytes - Viewed (0) -
tensorflow/c/eager/gradients_test.cc
string message = "This is the way!"; s = SetAttrString(check_numerics_op.get(), "message", message.data(), message.length(), &forward_op); ASSERT_EQ(errors::OK, s.code()) << s.message(); int num_retvals = 1; std::vector<AbstractTensorHandle*> outputs(1); GradientRegistry registry; s = RegisterGradients(®istry); ASSERT_EQ(errors::OK, s.code()) << s.message();
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/experimental/gradients/tape/tape_operation.cc
} Status TapeOperation::SetAttrShapeList(const char* attr_name, const int64_t** dims, const int* num_dims, int num_values) { std::unique_ptr<TensorShapeProto[]> proto(new TensorShapeProto[num_values]); for (int i = 0; i < num_values; ++i) { const auto num_dims_i = num_dims[i]; if (num_dims_i > TensorShape::MaxDimensions()) { return errors::InvalidArgument(
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Tue Jun 07 01:53:35 GMT 2022 - 9K bytes - Viewed (1)