Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for attr_in (0.32 sec)

  1. 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)
  2. 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)
Back to top