Searched refs:kDynamicRNNOpName (Results 1 – 8 of 8) sorted by relevance
42 if (op_name != kDynamicRNNOpName) { in Process()
171 if (op_name == kDynamicRNNOpName && i == kNonePlaceholderIdx) { in LaunchKernel()
420 static std::set<std::string> node_set = {kDynamicRNNOpName, kDynamicGRUV2OpName}; in IsPlaceHolderInput()
324 if (op_name == kDynamicRNNOpName && input_ptr->name() == "seq_length") { in GenInputDescJson()
281 constexpr auto kDynamicRNNOpName = "DynamicRNN"; variable
532 …if ((op_name == kDynamicRNNOpName || op_name == kDynamicGRUV2OpName) && input_origin_type == kMeta… in InitCommonNodeInputs()
1139 if (op_name == kDynamicRNNOpName && i == none_placeholder_index) { in GenLaunchArgs()
2391 if (op_name == kDynamicRNNOpName && i == none_placeholder_index) { in CacheAddrForKernel()