Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for TFE_OpSetAttrType (0.05 sec)

  2. github.com/tensorflow/tensorflow

    tensorflow/c/c_api_experimental_test.cc 

      TFE_Op* reshape_op = TFE_NewOp(tfe_context_, "Reshape", status_);
  CHECK_EQ(TF_OK, TF_GetCode(status_)) << TF_Message(status_);
  TFE_OpSetAttrType(reshape_op, "T", TF_FLOAT);
  TFE_OpSetAttrType(reshape_op, "Tshape", TF_INT32);
  CheckOutputShapes(reshape_op,
                    /* input_shapes*/ {unknown_shape(), unknown_shape()},
    Registered: Tue Sep 09 12:39:10 UTC 2025
    - Last Modified: Tue Jan 17 22:27:52 UTC 2023
    - 13.1K bytes
    - Viewed (0)
Back to top