HomeSort by relevance Sort by last modified time
    Searched refs:type_constraints (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/compiler/tf2xla/
xla_op_registry.cc 181 for (const auto& constraint : op_registration->type_constraints) {
216 auto constraint_it = op_registration->type_constraints.find(type_attr);
217 const std::set<DataType>* type_constraints = local
218 constraint_it != op_registration->type_constraints.end()
230 if (type_constraints != nullptr &&
231 type_constraints->find(dtype) == type_constraints->end()) {
350 registration_->type_constraints[attr_name.ToString()];
358 registration_->type_constraints[attr_name.ToString()];
xla_op_registry.h 182 std::unordered_map<string, std::set<DataType>> type_constraints; member in struct:tensorflow::XlaOpRegistry::OpRegistration

Completed in 137 milliseconds