- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 49 for type (0.14 sec)
-
tensorflow/c/c_api.cc
const auto* shapes_and_types = ic->output_handle_shapes_and_types(output.index); if (shapes_and_types == nullptr) return nullptr; for (const auto& p : *shapes_and_types) { auto* out_shape_and_type = handle_data.add_shape_and_type(); ic->ShapeHandleToProto(p.shape, out_shape_and_type->mutable_shape()); out_shape_and_type->set_dtype(p.dtype);
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/dlpack_test.cc
EXPECT_EQ(dltensor_out->device.device_type, dltensor_in->device.device_type); EXPECT_EQ(dltensor_out->device.device_id, dltensor_in->device.device_id); EXPECT_EQ(dltensor_out->ndim, dltensor_in->ndim); EXPECT_EQ(dltensor_out->dtype.code, dltensor_in->dtype.code); EXPECT_EQ(dltensor_out->dtype.bits, dltensor_in->dtype.bits); EXPECT_EQ(dltensor_out->dtype.lanes, dltensor_in->dtype.lanes); for (int i = 0; i < dltensor_in->ndim; ++i) {
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Fri Jun 30 03:04:46 GMT 2023 - 4.4K bytes - Viewed (0) -
tensorflow/c/eager/c_api_unified_experimental.cc
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, /*raw_device_name=*/nullptr)); } void TF_AbstractOpSetOpName(TF_AbstractOp* op, const char* const op_name,
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/dlpack.cc
owner->reference.Unref(); delete owner; } // Converts TF_DATAType to DLPack data type. DLDataType GetDlDataType(TF_DataType data_type, TF_Status* status) { DLDataType dtype; dtype.lanes = 1; dtype.bits = TF_DataTypeSize(data_type) * 8; switch (data_type) { case TF_DataType::TF_BOOL: dtype.code = DLDataTypeCode::kDLBool; break; case TF_DataType::TF_HALF: case TF_DataType::TF_FLOAT:
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 12.8K bytes - Viewed (0) -
tensorflow/c/eager/gradients.cc
input_ids[i] = ToId(inputs[i]); input_dtypes[i] = inputs[i]->DataType(); } std::vector<TapeTensor> tape_tensors; tape_tensors.reserve(outputs.size()); for (auto t : outputs) { tape_tensors.push_back(TapeTensor(t)); } GradientTape::RecordOperation( op_name, tape_tensors, input_ids, input_dtypes, [gradient_function]() -> GradientFunction* { return gradient_function; },
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/experimental/grappler/grappler.cc
TF_RETURN_IF_ERROR(ValidateTPOptimizerConfigs(optimizer_configs)); CGraphOptimizerRegister( [=]() { return new CGraphOptimizer(optimizer, params.device_type); }, optimizer_configs, params.device_type); return absl::OkStatus(); } } // namespace grappler } // namespace tensorflow void TF_GetNodesToPreserveListSize(const TF_GrapplerItem* item, int* num_values,
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Wed Sep 06 19:12:29 GMT 2023 - 15K bytes - Viewed (1) -
tensorflow/c/c_test_util.cc
} return found_dtype && found_shape; } bool IsScalarConst(const tensorflow::NodeDef& node_def, int v) { if (node_def.op() != "Const" || node_def.name() != "scalar") { return false; } bool found_dtype = false; bool found_value = false; for (const auto& attr : node_def.attr()) { if (attr.first == "dtype") { if (attr.second.type() == tensorflow::DT_INT32) { found_dtype = true;
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/eager/custom_device_test.cc
executed = false; TFE_Execute(matmul.get(), &retval, &num_retvals, status.get()); EXPECT_FALSE(executed); ASSERT_FALSE(TF_GetCode(status.get()) == TF_OK); // Custom devices can refuse to do type-based dispatch (as hcustom1 is // configured to do) matmul.reset(MatMulOp(context.get(), hcustom1.get(), hcpu.get())); num_retvals = 1; executed = false;
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Thu Aug 27 23:39:24 GMT 2020 - 18.4K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/array_grad_test.cc
#include "tensorflow/c/eager/c_api_unified_experimental_internal.h" #include "tensorflow/c/eager/unified_api_testutil.h" #include "tensorflow/c/experimental/gradients/grad_test_helper.h" #include "tensorflow/c/experimental/gradients/tape/tape_context.h" #include "tensorflow/c/experimental/ops/array_ops.h" #include "tensorflow/c/tf_status_helper.h" #include "tensorflow/core/platform/tensor_float_32_utils.h" #include "tensorflow/core/platform/test.h"
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/eager/immediate_execution_tensor_handle.cc
const char* device_name = DeviceName(&s); if (!s.ok()) { device_name = "<error fetching device name>"; } return absl::StrCat("TensorHandle(", value_string, ", shape=", shape_string, ", dtype=", DataType_Name(DataType()), ", device=\"", device_name, "\")"); } Status ImmediateExecutionTensorHandle::SummarizeValue( std::string& summary) const { Status status;
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 2.1K bytes - Viewed (0)