- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for MlirTensor (0.45 sec)
-
tensorflow/compiler/mlir/tensorflow/c/c_api_unified_experimental_mlir.cc
Status s = tensorflow::ConvertDataType(dtype, builder, type); if (s.ok()) *type = UnrankedTensorType::get(*type); return s; } class MlirTensor : public TracingTensorHandle { public: explicit MlirTensor(Value value) : TracingTensorHandle(kMlir), value_(value) {} tensorflow::DataType DataType() const override { tensorflow::DataType type;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 28.2K bytes - Viewed (0) -
tensorflow/c/eager/unified_api_test.cc
} return absl::OkStatus(); } TEST_P(UnifiedAPI, TestTensorShapeScalar) { if (UseFunction() && UseMlir()) { // TODO(b/173074167): Remove this. GTEST_SKIP() << "MlirTensor::Shape is not implemented yet."; } AbstractContextPtr ctx; { AbstractContext* ctx_raw = nullptr; Status s = BuildImmediateExecutionContext(std::get<1>(GetParam()), &ctx_raw);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Feb 27 13:57:45 UTC 2024 - 6.7K bytes - Viewed (0) -
tensorflow/c/eager/c_api_unified_experimental_internal.h
// ============================================================================= // Represents either a MlirTensor or a GraphTensor. // This base class does not expose any public methods other than to distinguish // which subclass it actually is. The user is responsible to use the right // type of AbstractTensor in their context (do not pass an MlirTensor to a // GraphContext and vice-versa). class TracingTensorHandle : public AbstractTensorHandle {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Nov 13 22:20:40 UTC 2020 - 5.2K bytes - Viewed (0)