- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for attr_in (0.08 sec)
-
tensorflow/c/kernels_test.cc
<< i; } TF_DeleteStatus(status); return static_cast<void*>(s); }; AttrValue v; std::string attr_in[] = {"bugs", "bunny", "duck"}; SetAttrValue(gtl::ArraySlice<std::string>(attr_in, 3), &v); CreateAndCallKernelWithAttr(my_create_func, "TestKernelAttrStringList", v); } TEST_F(TestKernelAttr, Tensor) { struct TensorProtoHelpers { public:
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 50.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc
tensorflow::Tensor tensor(list->element_dtype, tensorflow::TensorShape(result_shape)); auto attr_or = tensorflow::ConvertTensor(tensor, &rewriter); if (!attr_or.ok()) return failure(); rewriter.replaceOpWithNewOp<TF::ConstOp>(op, attr_or.value()); return success(); } // Extract individual tensor list element and combine them using the tf.Pack // op.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 70.7K bytes - Viewed (0)