Home
last modified time | relevance | path

Searched refs:compile_time_constant_inputs (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_op_registry.cc101 if (x.compile_time_constant_inputs != y.compile_time_constant_inputs) { in IsCompatible()
384 return &it->second.front()->compile_time_constant_inputs; in CompileTimeConstantInputArgNames()
398 const std::unordered_set<string>* compile_time_constant_inputs; in CompileTimeConstantInputs() local
405 compile_time_constant_inputs = &compile_time_constant_inputs_from_attr; in CompileTimeConstantInputs()
407 compile_time_constant_inputs = in CompileTimeConstantInputs()
409 if (compile_time_constant_inputs->empty()) { in CompileTimeConstantInputs()
417 << absl::StrJoin(*compile_time_constant_inputs, ", "); in CompileTimeConstantInputs()
419 for (const string& input : *compile_time_constant_inputs) { in CompileTimeConstantInputs()
548 registration_->compile_time_constant_inputs.emplace(input_name); in CompileTimeConstantInput()
Dxla_op_registry.h276 std::unordered_set<string> compile_time_constant_inputs; member
/external/tensorflow/tensorflow/compiler/jit/
Dincrease_dynamism_for_auto_jit_pass_test.cc133 std::vector<string> compile_time_constant_inputs; in TEST() local
134 compile_time_constant_inputs.push_back("size"); in TEST()
137 Attr(kXlaCompileTimeConstantInputsAttr, compile_time_constant_inputs), in TEST()