Searched refs:num_broadcast_args_ (Results 1 – 1 of 1) sorted by relevance
154 num_broadcast_args_(num_broadcast_args), in ParameterInfo()165 int64 NumBroadcastArgs() const { return num_broadcast_args_; } in NumBroadcastArgs()185 num_broadcast_args_); in IsBroadcastArg()189 return index >= (num_per_replica_args_ + num_broadcast_args_) && in IsVariableArg()191 num_broadcast_args_ + num_variables_); in IsVariableArg()196 num_broadcast_args_ + num_variables_) && in IsConstantArg()198 num_broadcast_args_ + num_variables_ + in IsConstantArg()205 num_broadcast_args_ + num_variables_ + num_guaranteed_constants_; in NumInputsFromHost()211 num_broadcast_args_ + num_variables_ + num_guaranteed_constants_; in NumInputsToEachReplica()240 int64 num_broadcast_args_ = 0; variable