Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_pass.h149 int64 num_variables, int64 num_guaranteed_constants, in ParameterInfo() argument
156 num_guaranteed_constants_(num_guaranteed_constants), in ParameterInfo()
Dencapsulate_tpu_computations_pass.cc2676 const int num_guaranteed_constants = in BuildTPUReplicateOps() local
2678 TF_RET_CHECK(num_guaranteed_constants >= 0); in BuildTPUReplicateOps()
2685 << " num_guaranteed_constants: " << num_guaranteed_constants in BuildTPUReplicateOps()
2690 num_variables + num_guaranteed_constants); in BuildTPUReplicateOps()
2695 num_guaranteed_constants + num_variables; in BuildTPUReplicateOps()
2760 DataTypeVector guaranteed_constant_types(num_guaranteed_constants); in BuildTPUReplicateOps()
2761 for (int i = 0; i < num_guaranteed_constants; ++i) { in BuildTPUReplicateOps()
Ddistributed_tpu_rewrite_pass.cc2355 const int num_guaranteed_constants = guaranteed_constant_nodes.size(); in BuildCompileNode() local
2356 const int guaranteed_const_start_index = num_args - num_guaranteed_constants; in BuildCompileNode()
2416 constant_arg_types.reserve(num_guaranteed_constants); in BuildCompileNode()
2417 for (int i = 0; i < num_guaranteed_constants; ++i) { in BuildCompileNode()
2450 for (int i = 0; i < num_guaranteed_constants; ++i) { in BuildCompileNode()