Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for XTimesY (0.06 sec)

  1. tensorflow/compiler/jit/xla_kernel_creator_test.cc

      return v;
    }
    
    TEST_F(XlaKernelCreatorTest, OneFloatOneResourceArgument) {
      FunctionDef fdef = XTimesY();
      (*fdef.mutable_attr())["_XlaMustCompile"] = BoolAttr(true);
      Init({fdef});
      XlaKernelCreator xla_kernel_creator;
      auto callsite =
          ToNodeProperties(R"pb(
            name: 'XTimesY' op: 'XTimesY' input: 'a' input: 'b'
          )pb");
      (*(callsite->node_def.mutable_attr()))["_XlaMustCompile"] = BoolAttr(true);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 16 01:39:55 UTC 2023
    - 5.7K bytes
    - Viewed (0)
Back to top