- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for DataTypeIsInteger (0.29 sec)
-
tensorflow/c/experimental/gradients/math_grad.cc
AbstractTensorHandle** output, const char* name) { auto dtype = input->DataType(); if (DataTypeIsFloating(BaseType(dtype)) || DataTypeIsInteger(BaseType(dtype))) { return tensorflow::ops::Identity(ctx, input, output, name); } else if (!DataTypeIsComplex(BaseType(dtype)) && BaseType(dtype) != DT_VARIANT) { return errors::InvalidArgument(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 28 13:53:47 UTC 2024 - 15.2K bytes - Viewed (0) -
tensorflow/compiler/jit/mark_for_compilation_pass.cc
} if (n->type_string() != "AssignAddVariableOp" && n->type_string() != "AssignSubVariableOp") { return false; } DataType dtype; if (!TryGetNodeAttr(n->def(), "dtype", &dtype) || !DataTypeIsInteger(dtype)) { return false; } Node* const_input = nullptr; for (const Edge* e : n->in_edges()) { if (!e->IsControlEdge() && e->src()->IsConstant()) { const_input = e->src();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 12:19:41 UTC 2024 - 85.3K bytes - Viewed (0)