Search Options

Display Count
Sort
Preferred Language
Advanced Search

Results 1 - 2 of 2 for TFE_OpSetAttrBool (0.09 seconds)

The search processing time has exceeded the limit. The displayed results may be partial.

  1. tensorflow/c/eager/c_api_test_util.cc

      TFE_OpAddInput(op, input, status);
      CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
      TFE_OpAddInput(op, axis, status);
      CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
      TFE_OpSetAttrBool(op, "keep_dims", 1);
      TFE_OpSetAttrType(op, "Tidx", TF_INT32);
      TF_DeleteStatus(status);
      TFE_OpSetAttrType(op, "T", TFE_TensorHandleDataType(input));
    
      return op;
    }
    
    Created: Tue Apr 07 12:39:13 GMT 2026
    - Last Modified: Thu Oct 09 05:56:18 GMT 2025
    - 23.4K bytes
    - Click Count (0)
  2. tensorflow/c/eager/c_api_test.cc

            serialized_config.length());
      }
    
      if (use_tfrt) {
        // Set some test-only graph compiler options.
        TFE_OpSetAttrBool(op, "TFRT_TEST_enable_native_ops", false);
        TFE_OpSetAttrBool(op, "TFRT_TEST_enable_grappler", enable_grappler);
      }
    
      ASSERT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
    
      TFE_OpAddInput(op, m, status);
    Created: Tue Apr 07 12:39:13 GMT 2026
    - Last Modified: Thu Oct 09 05:56:18 GMT 2025
    - 94.6K bytes
    - Click Count (0)
Back to Top